Skip to content

Commit 1d8f0db

Browse files
Provisory url for the Swin model
Signed-off-by: João Lucas de Sousa Almeida <[email protected]>
1 parent 59fb323 commit 1d8f0db

File tree

1 file changed

+4
-13
lines changed

1 file changed

+4
-13
lines changed

src/terratorch/models/backbones/prithvi_swin.py

+4-13
Original file line numberDiff line numberDiff line change
@@ -36,24 +36,15 @@ def _cfg(file: Path = "", **kwargs) -> dict:
3636
**kwargs,
3737
}
3838

39-
4039
default_cfgs = generate_default_cfgs(
4140
{
42-
# us trained model
43-
"prithvi_swin_90_us": _cfg(
44-
file="/dccstor/geofm-finetuning/pretrain_ckpts/swin_weights/2023-07-24_14-06-22/epoch-99-loss-0.1632_mmseg.pt"
45-
),
46-
# global models
47-
"prithvi_swin_B": _cfg(
48-
file="/dccstor/geofm-finetuning/swin_weights/2023-10-15_16-08-35/epoch-100-loss-0.0923.pt"
49-
),
50-
"prithvi_swin_L": _cfg(
51-
file="/dccstor/geofm-finetuning/swin_weights/2023-10-25_14-59-41/epoch-94-loss-0.0918.pt"
52-
),
41+
"prithvi_swin_90_us": {
42+
"hf_hub_id": "ibm-nasa-geospatial/Prithvi-100M",
43+
"hf_hub_filename": "Prithvi_100M.pt"
44+
}
5345
}
5446
)
5547

56-
5748
def convert_weights_swin2mmseg(ckpt):
5849
# from https://github.com/open-mmlab/mmsegmentation/blob/main/tools/model_converters/swin2mmseg.py
5950
new_ckpt = OrderedDict()

0 commit comments

Comments
 (0)