ibm-nasa-geospatial
/

Prithvi-EO-1.0-100M-sen1floods11

Image Segmentation

Foundation model

Model card Files Files and versions

carlosgomes98 commited on Nov 6, 2023

Commit

220f62f

·

1 Parent(s): f0220fc

Pretrained in EncoderDecoder

Files changed (1) hide show

sen1floods11_Prithvi_100M.py +1 -1

sen1floods11_Prithvi_100M.py CHANGED Viewed

@@ -224,9 +224,9 @@ ce_weights = [0.3, 0.7]
 model = dict(
     type="TemporalEncoderDecoder",
     frozen_backbone=False,
     backbone=dict(
         type="TemporalViTEncoder",
-        pretrained=pretrained_weights_path,
         img_size=img_size,
         patch_size=patch_size,
         num_frames=num_frames,

 model = dict(
     type="TemporalEncoderDecoder",
     frozen_backbone=False,
+    pretrained=pretrained_weights_path,
     backbone=dict(
         type="TemporalViTEncoder",
         img_size=img_size,
         patch_size=patch_size,
         num_frames=num_frames,