diff --git a/examples/confs/sen1floods11_vit.yaml b/examples/confs/sen1floods11_vit.yaml index 806d5138..9e80b10f 100644 --- 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 index 6cdcc3d9..62180b56 100644 --- 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 diff --git a/src/terratorch/models/prithvi_model_factory.py b/src/terratorch/models/prithvi_model_factory.py index ad424084..a84a35a8 100644 --- a/src/terratorch/models/prithvi_model_factory.py +++ b/src/terratorch/models/prithvi_model_factory.py @@ -32,7 +32,7 @@ def build_model( backbone: str | nn.Module, decoder: str | nn.Module, bands: list[HLSBands | int], - in_channels: int = int | None, # this should be removed, can be derived from bands. But it is a breaking change + in_channels: int | None = None, # this should be removed, can be derived from bands. But it is a breaking change num_classes: int | None = None, pretrained: bool = True, # noqa: FBT001, FBT002 num_frames: int = 1,