Update test_results.py
This commit is contained in:
parent
bf70e431b3
commit
ea74906878
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue