How to disable auto log when loop in Dataloader

data
1

Hi, I’m new to Pytorch. When I use for loop in Dataloder, it always auto log lists like below.

for image, mask in trainloader:  
  break
  

print(f"batch image shape : {image.shape}")
print(f"batch in mask shape : {mask.shape}")

image

Is there a way to disable that log lists?

2

PyTorch does not show any logging info from the Dataset so I guess your (custom) dataset might print it and you should check the implementation.