Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "b0a2eb31ee2d4384502d52422fec18044347ce84"
Commit 3b1703e3 authored by WenmuZhou's avatar WenmuZhou
Browse files

add dataset len check

parent b9d3e8ee
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tools/train.py
View file @ 3b1703e3
...@@ -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% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment