add dataset len check

2026-09-24 23:33:08 +08:00 · 2021-02-04 12:01:56 +08:00
parent b9d3e8eeec
commit 3b1703e3eb
1 changed files with 1 additions and 0 deletions
@@ -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)