SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Add nn.Module as superclass (#4533)

Browse Source
This commit is contained in:
Sho Arora
2020-05-25 12:29:33 -07:00
committed by GitHub
parent 8f7c1c7672
commit adab7f8332
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/modeling_mmbt.py
View File

@@ -149,7 +149,7 @@ MMBT_INPUTS_DOCSTRING = r""" Inputs:
MMBT_START_DOCSTRING, MMBT_START_DOCSTRING,
MMBT_INPUTS_DOCSTRING, MMBT_INPUTS_DOCSTRING,
) )
class MMBTModel(ModuleUtilsMixin): class MMBTModel(nn.Module, ModuleUtilsMixin):
r""" r"""
Outputs: `Tuple` comprising various elements depending on the configuration (config) and inputs: Outputs: `Tuple` comprising various elements depending on the configuration (config) and inputs:
**last_hidden_state**: ``torch.FloatTensor`` of shape ``(batch_size, sequence_length, hidden_size)`` **last_hidden_state**: ``torch.FloatTensor`` of shape ``(batch_size, sequence_length, hidden_size)``