提交 868ae8af 编写于 作者: littletomatodonkey's avatar littletomatodonkey

fix rec yaml

上级 558c2dd4
...@@ -4,7 +4,6 @@ Global: ...@@ -4,7 +4,6 @@ Global:
pretrained_model: null pretrained_model: null
output_dir: "./output/" output_dir: "./output/"
device: "gpu" device: "gpu"
class_num: 5013
save_interval: 1 save_interval: 1
eval_mode: "retrieval" eval_mode: "retrieval"
eval_during_train: True eval_during_train: True
......
...@@ -4,7 +4,6 @@ Global: ...@@ -4,7 +4,6 @@ Global:
pretrained_model: null pretrained_model: null
output_dir: "./output/" output_dir: "./output/"
device: "gpu" device: "gpu"
class_num: 3000
save_interval: 1 save_interval: 1
eval_during_train: True eval_during_train: True
eval_interval: 1 eval_interval: 1
......
...@@ -4,7 +4,6 @@ Global: ...@@ -4,7 +4,6 @@ Global:
pretrained_model: null pretrained_model: null
output_dir: ./output/ output_dir: ./output/
device: gpu device: gpu
class_num: 50030
save_interval: 1 save_interval: 1
eval_during_train: True eval_during_train: True
eval_interval: 10 eval_interval: 10
......
...@@ -6,7 +6,6 @@ Global: ...@@ -6,7 +6,6 @@ Global:
pretrained_model: product_ResNet50_vd_Aliproduct_v1.0_pretrained pretrained_model: product_ResNet50_vd_Aliproduct_v1.0_pretrained
output_dir: ./output/ output_dir: ./output/
device: gpu device: gpu
class_num: 3997
save_interval: 10 save_interval: 10
eval_during_train: True eval_during_train: True
eval_interval: 10 eval_interval: 10
......
...@@ -6,7 +6,6 @@ Global: ...@@ -6,7 +6,6 @@ Global:
pretrained_model: product_ResNet50_vd_Aliproduct_v1.0_pretrained pretrained_model: product_ResNet50_vd_Aliproduct_v1.0_pretrained
output_dir: ./output/ output_dir: ./output/
device: gpu device: gpu
class_num: 11319
save_interval: 10 save_interval: 10
eval_during_train: True eval_during_train: True
eval_interval: 10 eval_interval: 10
......
...@@ -4,7 +4,6 @@ Global: ...@@ -4,7 +4,6 @@ Global:
pretrained_model: null pretrained_model: null
output_dir: "./output/" output_dir: "./output/"
device: "gpu" device: "gpu"
class_num: 431
save_interval: 1 save_interval: 1
eval_during_train: True eval_during_train: True
eval_interval: 1 eval_interval: 1
......
...@@ -4,7 +4,6 @@ Global: ...@@ -4,7 +4,6 @@ Global:
pretrained_model: null pretrained_model: null
output_dir: "./output/" output_dir: "./output/"
device: "gpu" device: "gpu"
class_num: 30671
save_interval: 1 save_interval: 1
eval_during_train: True eval_during_train: True
eval_interval: 1 eval_interval: 1
......
...@@ -4,7 +4,6 @@ Global: ...@@ -4,7 +4,6 @@ Global:
pretrained_model: null pretrained_model: null
output_dir: ./output/ output_dir: ./output/
device: gpu device: gpu
class_num: 101
save_interval: 5 save_interval: 5
eval_during_train: True eval_during_train: True
eval_interval: 1 eval_interval: 1
......
...@@ -4,7 +4,6 @@ Global: ...@@ -4,7 +4,6 @@ Global:
pretrained_model: null pretrained_model: null
output_dir: ./output/ output_dir: ./output/
device: gpu device: gpu
class_num: 102
save_interval: 1 save_interval: 1
eval_during_train: True eval_during_train: True
eval_interval: 1 eval_interval: 1
...@@ -18,6 +17,7 @@ Global: ...@@ -18,6 +17,7 @@ Global:
# model architecture # model architecture
Arch: Arch:
name: MobileNetV3_large_x1_0 name: MobileNetV3_large_x1_0
class_num: 102
# loss function config for traing/eval process # loss function config for traing/eval process
Loss: Loss:
......
...@@ -4,7 +4,6 @@ Global: ...@@ -4,7 +4,6 @@ Global:
pretrained_model: null pretrained_model: null
output_dir: ./output/ output_dir: ./output/
device: cpu device: cpu
class_num: 102
save_interval: 1 save_interval: 1
eval_during_train: True eval_during_train: True
eval_interval: 1 eval_interval: 1
...@@ -18,6 +17,7 @@ Global: ...@@ -18,6 +17,7 @@ Global:
# model architecture # model architecture
Arch: Arch:
name: ShuffleNetV2_x0_25 name: ShuffleNetV2_x0_25
class_num: 102
# loss function config for traing/eval process # loss function config for traing/eval process
Loss: Loss:
......
...@@ -4,7 +4,6 @@ Global: ...@@ -4,7 +4,6 @@ Global:
pretrained_model: null pretrained_model: null
output_dir: ./output/ output_dir: ./output/
device: gpu device: gpu
class_num: 100
save_interval: 1 save_interval: 1
eval_during_train: True eval_during_train: True
eval_interval: 1 eval_interval: 1
...@@ -18,6 +17,7 @@ Global: ...@@ -18,6 +17,7 @@ Global:
# model architecture # model architecture
Arch: Arch:
name: MobileNetV3_large_x1_0 name: MobileNetV3_large_x1_0
class_num: 100
# loss function config for traing/eval process # loss function config for traing/eval process
Loss: Loss:
......
...@@ -4,7 +4,6 @@ Global: ...@@ -4,7 +4,6 @@ Global:
pretrained_model: null pretrained_model: null
output_dir: "./output/" output_dir: "./output/"
device: "gpu" device: "gpu"
class_num: 100
save_interval: 1 save_interval: 1
eval_during_train: True eval_during_train: True
eval_interval: 1 eval_interval: 1
...@@ -27,9 +26,11 @@ Arch: ...@@ -27,9 +26,11 @@ Arch:
models: models:
- Teacher: - Teacher:
name: ResNet50_vd name: ResNet50_vd
class_num: 100
pretrained: "./pretrained/best_model" pretrained: "./pretrained/best_model"
- Student: - Student:
name: MobileNetV3_large_x1_0 name: MobileNetV3_large_x1_0
class_num: 100
pretrained: True pretrained: True
infer_model_name: "Student" infer_model_name: "Student"
......
...@@ -4,7 +4,6 @@ Global: ...@@ -4,7 +4,6 @@ Global:
pretrained_model: null pretrained_model: null
output_dir: ./output/ output_dir: ./output/
device: gpu device: gpu
class_num: 100
save_interval: 1 save_interval: 1
eval_during_train: True eval_during_train: True
eval_interval: 1 eval_interval: 1
...@@ -18,6 +17,7 @@ Global: ...@@ -18,6 +17,7 @@ Global:
# model architecture # model architecture
Arch: Arch:
name: ResNet50_vd name: ResNet50_vd
class_num: 100
# loss function config for traing/eval process # loss function config for traing/eval process
Loss: Loss:
......
...@@ -4,7 +4,6 @@ Global: ...@@ -4,7 +4,6 @@ Global:
pretrained_model: null pretrained_model: null
output_dir: ./output/ output_dir: ./output/
device: gpu device: gpu
class_num: 100
save_interval: 1 save_interval: 1
eval_during_train: True eval_during_train: True
eval_interval: 1 eval_interval: 1
...@@ -18,6 +17,7 @@ Global: ...@@ -18,6 +17,7 @@ Global:
# model architecture # model architecture
Arch: Arch:
name: ResNet50_vd name: ResNet50_vd
class_num: 100
# loss function config for traing/eval process # loss function config for traing/eval process
Loss: Loss:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册