Update torchcallback.py

delve-team · Mar 6, 2020 · 3c55224 · 3c55224
1 parent f0452e9
commit 3c55224
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/delve/torchcallback.py b/delve/torchcallback.py
@@ -290,7 +290,6 @@ def _record_stat(self, activations_batch: torch.Tensor, lstm_ae: bool, layer: to
         if activations_batch.dim() == 4:  # conv layer (B x C x H x W)
             if self.interpolation_strategy is not None and (activations_batch.shape[3] > self.interpolation_downsampling or activations_batch.shape[2] > self.interpolation_downsampling):
                 activations_batch = interpolate(activations_batch, size=self.interpolation_downsampling, mode=self.interpolation_strategy)
-                print(activations_batch.shape)
             if self.conv_method == 'median':
                 shape = activations_batch.shape
                 reshaped_batch = activations_batch.reshape(shape[0], shape[1], shape[2] * shape[3])