argument not used

Signed-off-by: João Lucas de Sousa Almeida <[email protected]>
IBM · Jan 6, 2025 · e04e53e · e04e53e
1 parent dde31bb
commit e04e53e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/terratorch/models/scalar_output_model.py b/terratorch/models/scalar_output_model.py
@@ -120,7 +120,7 @@ def forward(self, x: torch.Tensor, **kwargs) -> ModelOutput:
         features = prepare(features)
 
         decoder_output = self.decoder([f.clone() for f in features])
-        decoder_output = self._crop_image_when_necessary(decoder_output, pad, input_size)
+        decoder_output = self._crop_image_when_necessary(decoder_output, input_size)
         mask = self.head(decoder_output)
 
         aux_outputs = {}