Skip to content

Commit d831cdb

Browse files
committed
remove another torchscript
1 parent ae94b8b commit d831cdb

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

libmultilabel/nn/networks/bert.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def __init__(
3434
hidden_dropout_prob=encoder_hidden_dropout,
3535
attention_probs_dropout_prob=encoder_attention_dropout,
3636
classifier_dropout=post_encoder_dropout,
37-
torchscript=True,
37+
# torchscript=True,
3838
)
3939

4040
def forward(self, input):

0 commit comments

Comments
 (0)