Merge pull request #99 from yurujaja/revert-98-fix/ss4eo_mae_optical_…

…checkpoint

Revert "fix checkpoint for ssl4eo_mae_optical (#98)"

pangaea/encoders/ssl4eo_mae_encoder.py

@@ -158,7 +158,7 @@ def forward(self, image):
 
     def load_encoder_weights(self, logger: Logger) -> None:
         checkpoint = torch.load(self.encoder_weights, map_location="cpu")
-        pretrained_model = checkpoint["state_dict"]
+        pretrained_model = checkpoint["model"]
 
         k = pretrained_model.keys()
         pretrained_encoder = {}