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

fix rec yaml

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