Update test_results.py

This commit is contained in:
shuyingsunshine21 2021-03-24 01:42:16 -07:00 committed by GitHub
parent bf70e431b3
commit ea74906878
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 3 deletions

View File

@ -26,8 +26,6 @@ from pytorch_lightning.core.step_result import Result
from pytorch_lightning.trainer.states import TrainerState
from tests.helpers import BoringDataModule, BoringModel
from tests.helpers.runif import RunIf
import os
import numpy
def _setup_ddp(rank, worldsize):
@ -52,7 +50,6 @@ def _ddp_test_fn(rank, worldsize, result_cls: Result):
def test_result_reduce_ddp():
"""Make sure result logging works with DDP"""
tutils.reset_seed()
os.environ["MKL_SERVICE_FORCE_INTEL"] = "1"
tutils.set_random_master_port()
worldsize = 2