aparler/PyNorch
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix distributed dataloader

Browse source
This commit is contained in:
lucasdelimanogueira 2024-05-27 15:09:07 -03:00
parent e3bb882d3c
commit 240e603c0c
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/test_distributed.py
View file

@ -9,7 +9,7 @@ from norch.norchvision import transforms as norch_transforms
from torchvision import transforms as torch_transforms from torchvision import transforms as torch_transforms
class TestDistributed(unittest.TestCase): class TestDistributed(unittest.TestCase):
"""def test_distributed_sampler_batch_1(self): def test_distributed_sampler_batch_1(self):
transforms = norch_transforms.Compose( transforms = norch_transforms.Compose(
[ [
norch_transforms.ToTensor(), norch_transforms.ToTensor(),
@ -48,7 +48,7 @@ class TestDistributed(unittest.TestCase):
break break
for label_norch, label_torch in zip(labels_norch, labels_torch): for label_norch, label_torch in zip(labels_norch, labels_torch):
self.assertTrue(utils.compare_torch(label_norch, label_torch))""" self.assertTrue(utils.compare_torch(label_norch, label_torch))
def test_distributed_sampler_batch_32_rank_2(self): def test_distributed_sampler_batch_32_rank_2(self):