Update util.py

fix issue yl4579#5
bharathraj-v · Oct 3, 2023 · ec36cc2 · ec36cc2
1 parent 43869c2
commit ec36cc2
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/Utils/PLBERT/util.py b/Utils/PLBERT/util.py
@@ -36,8 +36,7 @@ def load_plbert(log_dir):
         if name.startswith('encoder.'):
             name = name[8:] # remove `encoder.`
             new_state_dict[name] = v
-    if not hasattr(bert.embeddings, 'position_ids'):
-        del new_state_dict["embeddings.position_ids"]
-    bert.load_state_dict(new_state_dict)
+    del new_state_dict["embeddings.position_ids"]
+    bert.load_state_dict(new_state_dict, strict=False)
 
     return bert