Merge pull request #1 from shuyingsunshine21/shuyingsunshine21-checkpoint_consolidate
Update test_all_gather_grad.py
This commit is contained in:
commit
c5053da789
|
@ -44,7 +44,6 @@ def _test_all_gather_ddp(rank, world_size):
|
|||
@RunIf(skip_windows=True)
|
||||
def test_all_gather_ddp():
|
||||
world_size = 3
|
||||
os.environ["MKL_SERVICE_FORCE_INTEL"] = "1"
|
||||
torch.multiprocessing.spawn(_test_all_gather_ddp, args=(world_size, ), nprocs=world_size)
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue