debug push

estimator/models/patchfusion.py

@@ -61,17 +61,21 @@ class PatchFusion(BaselinePretrain, PyTorchModelHubMixin):
         """
         nn.Module.__init__(self)
         
+        print(config)
         if isinstance(config, ConfigDict):
             # convert a ConfigDict to a PretrainedConfig for hf saving
             config = PretrainedConfig.from_dict(config.to_dict())
             config.load_branch = True
+            print("test1")
         else:
             # used when loading patchfusion from hf model space
             config = PretrainedConfig.from_dict(ConfigDict(**config).to_dict())
             config.load_branch = False
             config.coarse_branch.pretrained_resource = None
             config.fine_branch.pretrained_resource = None
-            
+            print("test2")
+        
+        print(config.coarse_branch) 
         self.config = config
         
         self.min_depth = config.min_depth