small fix

2020-09-11 23:04:26 +08:00
parent 340a344e91
commit f70658eaed
2 changed files with 2 additions and 3 deletions
--- a/main.py
+++ b/main.py
@@ -32,8 +32,7 @@ def running(local_rank, config, task, backup_config=False, setup_output_dir=Fals

    if setup_output_dir and config.resume_from is None:
        if output_dir.exists():
-            assert len(list(output_dir.glob("events*"))) == 0
-            assert len(list(output_dir.glob("*.pt"))) == 0
+            assert len(list(output_dir.glob("events*"))) == 0, f"{output_dir} containers tensorboard event"
            if (output_dir / "train.log").exists() and idist.get_rank() == 0:
                (output_dir / "train.log").unlink()
        else: