We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ae94b8b commit d831cdbCopy full SHA for d831cdb
1 file changed
libmultilabel/nn/networks/bert.py
@@ -34,7 +34,7 @@ def __init__(
34
hidden_dropout_prob=encoder_hidden_dropout,
35
attention_probs_dropout_prob=encoder_attention_dropout,
36
classifier_dropout=post_encoder_dropout,
37
- torchscript=True,
+ # torchscript=True,
38
)
39
40
def forward(self, input):
0 commit comments