add note

2020-11-27 18:36:50 +00:00 · 2020-11-27 18:36:50 +00:00 · d5d64f0ff6
parent 1f1a20c45f
commit d5d64f0ff6
1 changed files with 1 additions and 1 deletions
--- 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)