set strict=false to avoid incompatible keys error during matching of …

…head layers
GreatRSingh · Dec 7, 2023 · 3ddf373 · 3ddf373
1 parent 1e47ffd
commit 3ddf373
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/deepchem/models/torch_models/modular.py b/deepchem/models/torch_models/modular.py
@@ -313,7 +313,8 @@ def load_from_pretrained(  # type: ignore
         if source_model is not None:
             for name, module in source_model.components.items():
                 if components is None or name in components:
-                    self.components[name].load_state_dict(module.state_dict())
+                    self.components[name].load_state_dict(module.state_dict(),
+                                                          strict=False)
             self.build_model()
 
         elif source_model is None: