Check None before going through iteration (huggingface#13250)

qqaatw · web-flow · commit d50649531f4a · 2021-08-30T08:18:51.000-04:00
* Check None before going through iteration

* Format
diff --git a/src/transformers/trainer.py b/src/transformers/trainer.py
@@ -1411,7 +1411,9 @@ def _load_state_dict_in_model(self, state_dict):
         load_result = self.model.load_state_dict(state_dict, strict=False)
 
         if len(load_result.missing_keys) != 0:
-            if set(load_result.missing_keys) == set(self.model._keys_to_ignore_on_save):
+            if self.model._keys_to_ignore_on_save is not None and set(load_result.missing_keys) == set(
+                self.model._keys_to_ignore_on_save
+            ):
                 self.model.tie_weights()
             else:
                 logger.warn(f"There were missing keys in the checkpoint model loaded: {load_result.missing_keys}.")
