Merge pull request #11 from IBM/feature/simplify_vit

Joao-L-S-Almeida · web-flow · commit c4afa03ab44a · 2024-05-27T09:24:05.000-03:00
simplify config files for vit sen1floods11
diff --git a/examples/confs/sen1floods11_vit.yaml b/examples/confs/sen1floods11_vit.yaml
@@ -79,7 +79,6 @@ model:
       decoder: FCNDecoder
       pretrained: true
       backbone: prithvi_vit_100
-      backbone_pretrain_img_size: 512
       decoder_channels: 256
       in_channels: 6
       bands:
diff --git a/examples/confs/sen1floods11_vit_local_ckpt.yaml b/examples/confs/sen1floods11_vit_local_ckpt.yaml
@@ -77,16 +77,6 @@ model:
       decoder: FCNDecoder
       pretrained: true
       backbone: prithvi_vit_100
-      backbone_pretrain_img_size: 512
-      backbone_embed_dim: 768
-      backbone_num_heads: 12 
-      backbone_decoder_depth: 8
-      backbone_decoder_num_heads: 16
-      backbone_decoder_embed_dim: 512
-      backbone_mlp_ratio: 4
-      backbone_patch_size: 16
-      backbone_in_chans: 6
-      backbone_num_frames: 1
       backbone_pretrained_cfg_overlay:
         file: examples/Prithvi_100M.pt
       decoder_channels: 256
