Merge pull request espnet#4329 from roshansh-cmu/wandb

Wandb Minor Fix for Model Resume
chintu619 · May 6, 2022 · 4f41a1a · 4f41a1a
2 parents a49cc60 + f0d7cc2
commit 4f41a1a
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: