add dataset len check

This commit is contained in:
WenmuZhou 2021-02-04 12:01:56 +08:00
parent b9d3e8eeec
commit 3b1703e3eb
1 changed files with 1 additions and 0 deletions

View File

@ -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)