assert resume is not None

2020-08-08 07:35:25 +08:00 · 2020-08-08 07:35:25 +08:00 · 7cf235781d
commit 7cf235781d
parent c520ce9501
1 changed files with 1 additions and 0 deletions
--- a/engine/cyclegan.py
+++ b/engine/cyclegan.py
@ -238,6 +238,7 @@ def run(task, config, logger):
            import traceback
            print(traceback.format_exc())
    elif task == "test":
+        assert config.resume_from is not None
        test_dataset = data.DATASET.build_with(config.data.test.dataset)
        logger.info(f"test with dataset:\n{test_dataset}")
        test_data_loader = idist.auto_dataloader(test_dataset, **config.data.test.dataloader)