Training and inference performance results are different on the validation dataset
This happens in PyTorch Lightning 2.1.2 and 2.1.3. Updating PyTorch Lightning to 2.2.1 solves this issue. We should update the requirements.txt and the README.
This happens in PyTorch Lightning 2.1.2 and 2.1.3. Updating PyTorch Lightning to 2.2.1 solves this issue. We should update the requirements.txt and the README.