I am new at
PyTorch Lightning. So, I am trying to learn the basics. When I am trying to run code from here (UNet implementation using PyTorch Lightning), I am getting an error
AttributeError: module 'pytorch_lightning' has no attribute 'data_loader'.
The code is fully the same. I am getting the error in this block,
@pl.data_loader def train_dataloader(self): return self.__dataloader()['train']
Could you tell me why I am getting this error and how can I solve this issue?