budui/raycv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

assert resume is not None

Browse Source
This commit is contained in:
Ray Wong 2020-08-08 07:35:25 +08:00
parent c520ce9501
commit 7cf235781d
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
engine/cyclegan.py
View File

@ -238,6 +238,7 @@ def run(task, config, logger):
import traceback import traceback
print(traceback.format_exc()) print(traceback.format_exc())
elif task == "test": elif task == "test":
assert config.resume_from is not None
test_dataset = data.DATASET.build_with(config.data.test.dataset) test_dataset = data.DATASET.build_with(config.data.test.dataset)
logger.info(f"test with dataset:\n{test_dataset}") logger.info(f"test with dataset:\n{test_dataset}")
test_data_loader = idist.auto_dataloader(test_dataset, **config.data.test.dataloader) test_data_loader = idist.auto_dataloader(test_dataset, **config.data.test.dataloader)