Skip to content

Commit

Permalink
indent
Browse files Browse the repository at this point in the history
Signed-off-by: Joao Lucas de Sousa Almeida <[email protected]>
  • Loading branch information
Joao-L-S-Almeida committed Jan 24, 2025
1 parent b0a4780 commit 0dc1e95
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion terratorch/models/backbones/select_patch_embed_weights.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
def patch_embed_weights_are_compatible(model_patch_embed: torch.Tensor, checkpoint_patch_embed: torch.Tensor) -> bool:
# check all dimensions are the same except for channel dimension
if len(model_patch_embed.shape) != len(checkpoint_patch_embed.shape):
return False
return False

model_shape = [model_patch_embed.shape[i] for i in range(len(model_patch_embed.shape)) if i != 1]
checkpoint_shape = [checkpoint_patch_embed.shape[i] for i in range(len(checkpoint_patch_embed.shape)) if i != 1]
Expand Down

0 comments on commit 0dc1e95

Please sign in to comment.