From 7cf235781d9d1966b3b458e323bb5ace26ed37d7 Mon Sep 17 00:00:00 2001 From: Ray Wong Date: Sat, 8 Aug 2020 07:35:25 +0800 Subject: [PATCH] assert resume is not None --- engine/cyclegan.py | 1 + 1 file changed, 1 insertion(+) diff --git a/engine/cyclegan.py b/engine/cyclegan.py index cacb6f9..80e38aa 100644 --- 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)