Merge pull request #11 from IBM/feature/simplify_vit

simplify config files for vit sen1floods11
IBM · May 27, 2024 · c4afa03 · c4afa03
2 parents 2866275 + 8730ade
commit c4afa03
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 11 deletions.
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