Refine error message (open-mmlab#1338)

yuexiayiren159 · Sep 4, 2023 · 8c934d2 · 8c934d2
1 parent 5671b53
commit 8c934d2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/examples/distributed_training.py b/examples/distributed_training.py
@@ -85,7 +85,7 @@ def main():
         collate_fn=dict(type='default_collate'))
     runner = Runner(
         model=MMResNet50(),
-        work_dir='./work_dir',
+        work_dir='./work_dirs',
         train_dataloader=train_dataloader,
         optim_wrapper=dict(optimizer=dict(type=SGD, lr=0.001, momentum=0.9)),
         train_cfg=dict(by_epoch=True, max_epochs=2, val_interval=1),

diff --git a/mmengine/utils/progressbar.py b/mmengine/utils/progressbar.py
@@ -115,7 +115,8 @@ def track_progress(func: Callable,
         task_num = len(tasks)
     else:
         raise TypeError(
-            '"tasks" must be an iterable object or a (iterator, int) tuple')
+            '"tasks" must be an iterable object or a (iterator, int) tuple, '
+            f'but got {type(tasks)}')
     prog_bar = ProgressBar(task_num, bar_width, file=file)
     results = []
     for task in tasks:
@@ -181,7 +182,8 @@ def track_parallel_progress(func: Callable,
         task_num = len(tasks)
     else:
         raise TypeError(
-            '"tasks" must be an iterable object or a (iterator, int) tuple')
+            '"tasks" must be an iterable object or a (iterator, int) tuple, '
+            f'but got {type(tasks)}')
     pool = init_pool(nproc, initializer, initargs)
     start = not skip_first
     task_num -= nproc * chunksize * int(skip_first)
@@ -232,7 +234,8 @@ def track_iter_progress(tasks: Union[list, Iterable],
         task_num = len(tasks)
     else:
         raise TypeError(
-            '"tasks" must be an iterable object or a (iterator, int) tuple')
+            '"tasks" must be an iterable object or a (iterator, int) tuple, '
+            f'but got {type(tasks)}')
     prog_bar = ProgressBar(task_num, bar_width, file=file)
     for task in tasks:
         yield task