From 240e603c0c7084b88d50b6db7e009695e5a0a0b1 Mon Sep 17 00:00:00 2001 From: lucasdelimanogueira Date: Mon, 27 May 2024 15:09:07 -0300 Subject: [PATCH] fix distributed dataloader --- tests/test_distributed.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_distributed.py b/tests/test_distributed.py index 342d87f..1b16a20 100644 --- a/tests/test_distributed.py +++ b/tests/test_distributed.py @@ -9,7 +9,7 @@ from norch.norchvision import transforms as norch_transforms from torchvision import transforms as torch_transforms class TestDistributed(unittest.TestCase): - """def test_distributed_sampler_batch_1(self): + def test_distributed_sampler_batch_1(self): transforms = norch_transforms.Compose( [ norch_transforms.ToTensor(), @@ -48,7 +48,7 @@ class TestDistributed(unittest.TestCase): break 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):