Commit 3b1703e3 authored by WenmuZhou's avatar WenmuZhou
Browse files

add dataset len check

parent b9d3e8ee
......@@ -54,6 +54,7 @@ def main(config, device, logger, vdl_writer):
logger.error(
'No Images in train dataset, please check annotation file and path in the configuration file'
)
return
if config['Eval']:
valid_dataloader = build_dataloader(config, 'Eval', device, logger)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment