Update ResNet50_vd_ssld_finetune.yaml
parent
5d2dd8f7cc
commit
3b1fad3fa8
|
@ -2,7 +2,7 @@ mode: 'train'
|
|||
ARCHITECTURE:
|
||||
name: 'ResNet50_vd'
|
||||
params:
|
||||
lr_mult_list: [0.1, 0.1, 0.2, 0.2, 0.3]
|
||||
lr_mult_list: [0.5, 0.5, 0.6, 0.6, 0.8]
|
||||
pretrained_model: "./pretrained/ResNet50_vd_ssld_pretrained"
|
||||
model_save_dir: "./output/"
|
||||
use_gpu: True
|
||||
|
|
Loading…
Reference in New Issue