Skip to content

Commit ee17c96

Browse files
Logging
Signed-off-by: João Lucas de Sousa Almeida <[email protected]>
1 parent 3fe4ac6 commit ee17c96

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

terratorch/tasks/segmentation_tasks.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -254,7 +254,7 @@ def validation_step(self, batch: Any, batch_idx: int, dataloader_idx: int = 0) -
254254
x = batch["image"]
255255
y = batch["mask"]
256256
model_output: ModelOutput = self(x)
257-
257+
print(x.shape, model_output.output.shape, y.shape)
258258
loss = self.val_loss_handler.compute_loss(model_output, y, self.criterion, self.aux_loss)
259259
self.val_loss_handler.log_loss(self.log, loss_dict=loss, batch_size=x.shape[0])
260260
y_hat_hard = to_segmentation_prediction(model_output)

0 commit comments

Comments
 (0)