提交 3b1703e3 编写于 作者: W WenmuZhou

add dataset len check

上级 b9d3e8ee
...@@ -54,6 +54,7 @@ def main(config, device, logger, vdl_writer): ...@@ -54,6 +54,7 @@ def main(config, device, logger, vdl_writer):
logger.error( logger.error(
'No Images in train dataset, please check annotation file and path in the configuration file' 'No Images in train dataset, please check annotation file and path in the configuration file'
) )
return
if config['Eval']: if config['Eval']:
valid_dataloader = build_dataloader(config, 'Eval', device, logger) valid_dataloader = build_dataloader(config, 'Eval', device, logger)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册