osanseviero
/

test_model_bertmesh

Model card Files Files and versions Community

Nick Sorros commited on Feb 21, 2022

Commit

895295d

·

1 Parent(s): b4da537

Fix num_labels

Files changed (1) hide show

model.py +1 -1

model.py CHANGED Viewed

@@ -19,7 +19,7 @@ class BertMesh(PreTrainedModel):
     def __init__(
         self,
         pretrained_model,
-        num_labels,
         hidden_size=512,
         dropout=0,
         multilabel_attention=False,

     def __init__(
         self,
         pretrained_model,
+        num_labels=28761,
         hidden_size=512,
         dropout=0,
         multilabel_attention=False,