Skip to content

Commit

Permalink
Fix patch_size lists in configs
Browse files Browse the repository at this point in the history
Signed-off-by: Benedikt Blumenstiel <[email protected]>
  • Loading branch information
blumenstiel committed Dec 3, 2024
1 parent ad2bfcd commit 93df86e
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions terratorch/models/backbones/prithvi_vit.py
Original file line number Diff line number Diff line change
Expand Up @@ -146,6 +146,8 @@ def _create_prithvi(

padding = kwargs.get("padding", "none")
patch_size = kwargs.get("patch_size", 16)
if isinstance(patch_size, list):
patch_size = patch_size[-1]

# Little hack because VIT does not support timm's features_only
encoder_only = kwargs.pop("features_only", False)
Expand Down

0 comments on commit 93df86e

Please sign in to comment.