From 1b8668d7b785fdc0a1b2372c33102ebd98d96614 Mon Sep 17 00:00:00 2001
From: "Eric.Lee2021" <305141918@qq.com>
Date: Tue, 23 Mar 2021 12:11:49 +0800
Subject: [PATCH] update

---
 train.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/train.py b/train.py
index 6e81b88..2987f8a 100644
--- a/train.py
+++ b/train.py
@@ -21,7 +21,6 @@ def train(data_cfg ='cfg/voc.data',accumulate = 1):
     num_workers = int(get_data_cfg['num_workers'])
     cfg_model = get_data_cfg['cfg_model']
     train_path = get_data_cfg['train']
-    valid_ptah = get_data_cfg['valid']
     num_classes = int(get_data_cfg['classes'])
     finetune_model = get_data_cfg['finetune_model']
     batch_size = int(get_data_cfg['batch_size'])
@@ -46,13 +45,12 @@ def train(data_cfg ='cfg/voc.data',accumulate = 1):
     print('model            : ',cfg_model)
     print('finetune_model   : ',finetune_model)
     print('train_path       : ',train_path)
-    print('valid_ptah       : ',valid_ptah)
     print('num_classes      : ',num_classes)
     print('batch_size       : ',batch_size)
     print('img_size         : ',img_size)
     print('multi_scale      : ',multi_scale)
-    print('lr0              : ',lr0)
     print('lr_step          : ',lr_step)
+    print('lr0              : ',lr0)
     # load model
     pattern_data_ = data_cfg.split("/")[-1:][0].replace(".data","")
     if "-tiny" in cfg_model:
-- 
GitLab