[Bug] Fix darts loop unittest (#218)

* fix_darts_loop

* fix darts_loop unittest
pull/227/head
humu789 2022-08-05 17:46:05 +08:00 committed by GitHub
parent e88e92cac4
commit e4305f3d0b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 2 deletions

View File

@ -150,7 +150,6 @@ class TestDartsLoop(TestCase):
self.assertEqual(loop.max_epochs, 3)
self.assertEqual(loop.max_iters, 12)
self.assertIsInstance(loop.mutator_dataloader, DataLoader)
self.assertEqual(loop.multi_loaders.num_loaders, 2)
# 2. DartsIterBasedTrainLoop
cfg = copy.deepcopy(self.iter_based_cfg)
@ -162,7 +161,6 @@ class TestDartsLoop(TestCase):
self.assertIsInstance(loop.runner, Runner)
self.assertEqual(loop.max_iters, 12)
self.assertIsInstance(loop.mutator_dataloader, DataLoader)
self.assertEqual(loop.multi_loaders.num_loaders, 2)
def test_run(self):
# 1. test DartsEpochBasedTrainLoop