If you want to refer to the old version of PyTorch Lightning, please use this version.
NOTE: There is a problem that len(train_loader) does not work in pytorch-lightning 1.5.2.
- After updating to 1.5.2, NotImplementedError:
train_dataloader· Discussion #10652 · PyTorchLightning/pytorch-lightning Lightning-AI/pytorch-lightning#10652