Revert "Update test_metric_result_integration.py"

This reverts commit bf70e431b3.
This commit is contained in:
shuyingsunshine21 2021-03-24 02:06:52 -07:00
parent 250d0aaaa2
commit 6c095b2370
1 changed files with 3 additions and 0 deletions

View File

@ -16,6 +16,8 @@ import torch
import torch.distributed as dist
import torch.multiprocessing as mp
from torchmetrics import Metric
import numpy
import os
import tests.helpers.utils as tutils
from pytorch_lightning.core.step_result import Result
@ -96,6 +98,7 @@ def test_result_reduce_ddp():
tutils.set_random_master_port()
worldsize = 2
os.environ["MKL_SERVICE_FORCE_INTEL"] = "1"
mp.spawn(_ddp_test_fn, args=(worldsize, ), nprocs=worldsize)