Prevent InferenceConfig.pretrained_model_path from being overwritten …

…to None
AdvancedPhotonSource · May 22, 2024 · 54dcb8c · 54dcb8c
1 parent d049b7b
commit 54dcb8c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/generic_trainer/configs.py b/generic_trainer/configs.py
@@ -241,6 +241,8 @@ def string_to_object(self, key, value):
         value = super().string_to_object(key, value)
         if key == 'model_save_dir':
             self.pretrained_model_path = os.path.join(value, 'best_model.pth')
+        if key == 'pretrained_model_path' and value is None:
+            value = self.pretrained_model_path
         return value