Update distributed_training.py

open-mmlab · Sep 1, 2023 · 9540f21 · 9540f21
1 parent 4eb168f
commit 9540f21
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/examples/distributed_training.py b/examples/distributed_training.py
@@ -77,14 +77,12 @@ def main():
         batch_size=32,
         dataset=train_set,
         sampler=dict(type='DefaultSampler', shuffle=True),
-        collate_fn=dict(type='default_collate'),
-        num_batch_per_epoch=5)
+        collate_fn=dict(type='default_collate'))
     val_dataloader = dict(
         batch_size=32,
         dataset=valid_set,
         sampler=dict(type='DefaultSampler', shuffle=False),
-        collate_fn=dict(type='default_collate'),
-        num_batch_per_epoch=5)
+        collate_fn=dict(type='default_collate'))
     runner = Runner(
         model=MMResNet50(),
         work_dir='./work_dir',