minor fix

NRCan · Dec 6, 2024 · c99d277 · c99d277
1 parent 2791f4f
commit c99d277
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/geo_deep_learning/models/dofa/dofa_seg.py b/geo_deep_learning/models/dofa/dofa_seg.py
@@ -429,7 +429,7 @@ def forward(self, x: Tensor):
         return outs
 
 def dofa_encoder_base(pretrained: bool = True, *args: Any, **kwargs: Any):
-    url: str = "https://huggingface.co/XShadow/DOFA/resolve/main/DOFA_ViT_base_e120.pth"
+    url: str = "https://huggingface.co/XShadow/DOFA/resolve/main/DOFA_ViT_base_e100.pth"
     model_dict = torch.hub.load_state_dict_from_url(url, progress=True, map_location='cpu')
     del model_dict["mask_token"]
     del model_dict["norm.weight"], model_dict["norm.bias"]