Philipp Wirth

I can reproduce with the attached file. Currently getting the following error: ``` Processing ./lightly-1.4.1.tar.gz Installing build dependencies ... done Getting requirements to build wheel ... error ERROR: Command errored...

HI @Chrispresso, thank you for reporting the issue! Could you please add the error output to the description?

Thanks for providing the full error trace! To me it looks like the error happens in torchvision's resnet: ``` File "/.venv/lib/python3.11/site-packages/torchvision/models/resnet.py", line 97, in forward out = self.bn2(out) File "/.venv/lib/python3.11/site-packages/torch/nn/modules/module.py",...

Curious, it could be due to the combination of multiple forward passes, batch norm, and distributed fabric. Did you already search for similar issues in https://github.com/Lightning-AI/pytorch-lightning/issues?

Thank you for figuring out the workaround! I will close this issue now. Feel free to reopen if anything else comes up.