提交 3388c47e 编写于 作者: littletomatodonkey's avatar littletomatodonkey 提交者: Tingquan Gao

fix config, test=document_fix

上级 692420fa
......@@ -3,6 +3,7 @@ ARCHITECTURE:
name: 'MobileNetV3_large_x1_0'
pretrained_model: "./pretrained/MobileNetV3_large_x1_0_pretrained"
model_save_dir: "./output/"
use_gpu: True
classes_num: 102
total_images: 1020
save_interval: 1
......
......@@ -6,6 +6,7 @@ pretrained_model:
- "./pretrained/flowers102_R50_vd_final/ppcls"
- "./pretrained/MobileNetV3_large_x1_0_pretrained"
model_save_dir: "./output/"
use_gpu: True
classes_num: 102
total_images: 7169
save_interval: 1
......
......@@ -4,6 +4,7 @@ ARCHITECTURE:
checkpoints: ""
pretrained_model: ""
use_gpu: True
model_save_dir: "./output/"
classes_num: 102
total_images: 1020
......
......@@ -3,6 +3,7 @@ ARCHITECTURE:
name: 'ResNet50_vd'
pretrained_model: "./pretrained/ResNet50_vd_pretrained"
model_save_dir: "./output/"
use_gpu: True
classes_num: 102
total_images: 1020
save_interval: 1
......
......@@ -5,6 +5,7 @@ ARCHITECTURE:
lr_mult_list: [0.1, 0.1, 0.2, 0.2, 0.3]
pretrained_model: "./pretrained/ResNet50_vd_ssld_pretrained"
model_save_dir: "./output/"
use_gpu: True
classes_num: 102
total_images: 1020
save_interval: 1
......
......@@ -5,6 +5,7 @@ ARCHITECTURE:
lr_mult_list: [0.1, 0.1, 0.2, 0.2, 0.3]
pretrained_model: "./pretrained/ResNet50_vd_ssld_pretrained"
model_save_dir: "./output/"
use_gpu: True
classes_num: 102
total_images: 1020
save_interval: 1
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册