jinaai
/

jina-bert-flash-implementation

🇪🇺 Region: EU

Model card Files Files and versions

Markus28 commited on Feb 27, 2024

Commit

a0c289c

·

1 Parent(s): 4c68a4c

strict=True for debugging

Files changed (1) hide show

modeling_bert.py +1 -1

modeling_bert.py CHANGED Viewed

@@ -329,7 +329,7 @@ class BertPreTrainedModel(nn.Module):
         """
         # Instantiate model.
         model = cls(config, *inputs, **kwargs)
-        load_return = model.load_state_dict(state_dict_from_pretrained(model_name), strict=False)
         logger.info(load_return)
         return model

         """
         # Instantiate model.
         model = cls(config, *inputs, **kwargs)
+        load_return = model.load_state_dict(state_dict_from_pretrained(model_name), strict=True)
         logger.info(load_return)
         return model