ibm-nasa-geospatial
/

Prithvi-100M-burn-scar

Foundation model

Model card Files Files and versions Community

carlosgomes98 commited on Nov 6, 2023

Commit

d8c0ec4

•

1 Parent(s): f2c0d71

Pretrained in EncoderDecoder

Files changed (1) hide show

burn_scars_Prithvi_100M.py +1 -1

burn_scars_Prithvi_100M.py CHANGED Viewed

@@ -173,9 +173,9 @@ norm_cfg = dict(type='BN', requires_grad=True)
 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,