diff --git a/benchmarks/test_parity.py b/benchmarks/test_parity.py index 433cfb0090..d026096a84 100644 --- a/benchmarks/test_parity.py +++ b/benchmarks/test_parity.py @@ -9,7 +9,7 @@ from pytorch_lightning import Trainer, seed_everything from tests.base.models import ParityModuleMNIST, ParityModuleRNN -# TODO: explore where the time leak comes from +# ParityModuleMNIST runs with num_workers=1 @pytest.mark.parametrize('cls_model,max_diff', [ (ParityModuleRNN, 0.05), (ParityModuleMNIST, 0.22)