From 3b1703e3eb30ef2a2610a34bd7f554f363113a74 Mon Sep 17 00:00:00 2001
From: WenmuZhou <zjwenmu@gmail.com>
Date: Thu, 4 Feb 2021 12:01:56 +0800
Subject: [PATCH] add dataset len check

---
 tools/train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tools/train.py b/tools/train.py
index a88d64c2..fab10b64 100755
--- a/tools/train.py
+++ b/tools/train.py
@@ -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)
-- 
GitLab