Wandb minor fix for model resume

chintu619 · Apr 30, 2022 · 79c071e · 79c071e
1 parent ffe7c58
commit 79c071e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/espnet2/tasks/abs_task.py b/espnet2/tasks/abs_task.py
@@ -1300,7 +1300,7 @@ def main_worker(cls, args: argparse.Namespace):
                         name=name,
                         dir=output_dir,
                         id=args.wandb_id,
-                        resume="allow",
+                        resume=args.resume,
                     )
                     wandb.config.update(args)
                 else: