diff --git a/train.py b/train.py index 6e81b88c0cbc7a01917d883afd60fdd1f743b44a..2987f8afed71af03211eeacab31ff3376e344587 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: