提交 1b8668d7 编写于 作者: Eric.Lee2021's avatar Eric.Lee2021 🚴🏻

update

上级 9efceef6
......@@ -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:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册