logging

[alikashlan10]

how to disable the logging in each step , this 👇

==================================================
=== Conv2d _conv_forward called ===
Input shape: torch.Size([2, 32, 128, 128])

==================================================
=== Conv2d _conv_forward called ===
Input shape: torch.Size([2, 32, 128, 128])

==================================================
=== Conv2d _conv_forward called ===
Input shape: torch.Size([2, 32, 128, 128])

==================================================
=== Conv2d _conv_forward called ===
Input shape: torch.Size([2, 32, 128, 128])

==================================================

[simon-donike]

Hi @alikashlan10 - thanks for reaching out. I can't seem to find any stray print statements or similar anywhere. If you share the config you're using, I can try to reproduce the issue.