未验证 提交 501a1791 编写于 作者: B Bin Lu 提交者: GitHub

Merge pull request #988 from littletomatodonkey/dev/fix_dim_cfg

fix yaml
...@@ -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: 1000
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: AlexNet name: AlexNet
class_num: 1000
# 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: 1000
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: DLA102 name: DLA102
class_num: 1000
# 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: 1000
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: DLA102x name: DLA102x
class_num: 1000
# 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: 1000
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: DLA102x2 name: DLA102x2
class_num: 1000
# 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: 1000
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: DLA169 name: DLA169
class_num: 1000
# 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: 1000
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: DLA34 name: DLA34
class_num: 1000
# 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: 1000
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: DLA46_c name: DLA46_c
class_num: 1000
# 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: 1000
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: DLA46x_c name: DLA46x_c
class_num: 1000
# 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: 1000
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: DLA60 name: DLA60
class_num: 1000
# 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: 1000
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: DLA60x name: DLA60x
class_num: 1000
# 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: 1000
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: DLA60x_c name: DLA60x_c
class_num: 1000
# 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: 1000
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: DPN107 name: DPN107
class_num: 1000
# 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: 1000
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: DPN131 name: DPN131
class_num: 1000
# 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: 1000
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: DPN68 name: DPN68
class_num: 1000
# 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: 1000
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: DPN92 name: DPN92
class_num: 1000
# 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: 1000
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: DPN98 name: DPN98
class_num: 1000
# 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: 1000
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: DarkNet53 name: DarkNet53
class_num: 1000
# 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: 1000
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 name: ResNet50
class_num: 1000
# 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: 1000
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 name: ResNet50
class_num: 1000
# 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: 1000
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 name: ResNet50
class_num: 1000
# 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: 1000
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 name: ResNet50
class_num: 1000
# 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: 1000
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 name: ResNet50
class_num: 1000
# 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: 1000
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 name: ResNet50
class_num: 1000
# 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: 1000
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 name: ResNet50
class_num: 1000
# 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: 1000
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 name: ResNet50
class_num: 1000
# 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: 1000
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 name: ResNet50
class_num: 1000
# 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: 1000
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: DeiT_base_distilled_patch16_224 name: DeiT_base_distilled_patch16_224
class_num: 1000
# 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: 1000
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: DeiT_base_distilled_patch16_384 name: DeiT_base_distilled_patch16_384
class_num: 1000
# 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: 1000
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: DeiT_base_patch16_224 name: DeiT_base_patch16_224
class_num: 1000
# 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: 1000
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: DeiT_base_patch16_384 name: DeiT_base_patch16_384
class_num: 1000
# 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: 1000
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: DeiT_small_distilled_patch16_224 name: DeiT_small_distilled_patch16_224
class_num: 1000
# 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: 1000
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: DeiT_small_patch16_224 name: DeiT_small_patch16_224
class_num: 1000
# 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: 1000
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: DeiT_tiny_distilled_patch16_224 name: DeiT_tiny_distilled_patch16_224
class_num: 1000
# 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: 1000
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: DeiT_tiny_patch16_224 name: DeiT_tiny_patch16_224
class_num: 1000
# 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: 1000
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: DenseNet121 name: DenseNet121
class_num: 1000
# 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: 1000
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: DenseNet161 name: DenseNet161
class_num: 1000
# 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: 1000
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: DenseNet169 name: DenseNet169
class_num: 1000
# 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: 1000
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: DenseNet201 name: DenseNet201
class_num: 1000
# 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: 1000
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: DenseNet264 name: DenseNet264
class_num: 1000
# 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: 1000
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: "DistillationModel" name: "DistillationModel"
class_num: 1000
# if not null, its lengths should be same as models # if not null, its lengths should be same as models
pretrained_list: pretrained_list:
# if not null, its lengths should be same as models # if not null, its lengths should be same as models
......
...@@ -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: 1000
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: EfficientNetB0 name: EfficientNetB0
class_num: 1000
# 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: 1000
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: GhostNet_x0_5 name: GhostNet_x0_5
class_num: 1000
# 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: 1000
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: GhostNet_x1_0 name: GhostNet_x1_0
class_num: 1000
# 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: 1000
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: GhostNet_x1_3 name: GhostNet_x1_3
class_num: 1000
# 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: 1000
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: HRNet_W18_C name: HRNet_W18_C
class_num: 1000
# 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: 1000
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: HRNet_W30_C name: HRNet_W30_C
class_num: 1000
# 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: 1000
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: HRNet_W32_C name: HRNet_W32_C
class_num: 1000
# 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: 1000
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: HRNet_W40_C name: HRNet_W40_C
class_num: 1000
# 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: 1000
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: HRNet_W44_C name: HRNet_W44_C
class_num: 1000
# 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: 1000
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: HRNet_W48_C name: HRNet_W48_C
class_num: 1000
# 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: 1000
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: HRNet_W64_C name: HRNet_W64_C
class_num: 1000
# 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: 1000
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: HarDNet39_ds name: HarDNet39_ds
class_num: 1000
# 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: 1000
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: HarDNet68 name: HarDNet68
class_num: 1000
# 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: 1000
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: HarDNet68_ds name: HarDNet68_ds
class_num: 1000
# 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: 1000
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: HarDNet85 name: HarDNet85
class_num: 1000
# 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: 1000
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: GoogLeNet name: GoogLeNet
class_num: 1000
# 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: 1000
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: InceptionV3 name: InceptionV3
class_num: 1000
# 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: 1000
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: InceptionV4 name: InceptionV4
class_num: 1000
# 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: 1000
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: LeViT_128 name: LeViT_128
class_num: 1000
# 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: 1000
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: LeViT_128S name: LeViT_128S
class_num: 1000
# 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: 1000
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: LeViT_192 name: LeViT_192
class_num: 1000
# 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: 1000
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: LeViT_256 name: LeViT_256
class_num: 1000
# 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: 1000
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: LeViT_384 name: LeViT_384
class_num: 1000
# 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: 1000
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:
to_static: False to_static: False
# model architecture # model architecture
class_num: 1000
Arch: Arch:
name: MobileNetV1 name: MobileNetV1
......
...@@ -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: 1000
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: MobileNetV1_x0_25 name: MobileNetV1_x0_25
class_num: 1000
# 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: 1000
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: MobileNetV1_x0_5 name: MobileNetV1_x0_5
class_num: 1000
# 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: 1000
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: MobileNetV1_x0_75 name: MobileNetV1_x0_75
class_num: 1000
# 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: 1000
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:
to_static: False to_static: False
# model architecture # model architecture
class_num: 1000
Arch: Arch:
name: MobileNetV2 name: MobileNetV2
......
...@@ -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: 1000
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: MobileNetV2_x0_25 name: MobileNetV2_x0_25
class_num: 1000
# 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: 1000
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: MobileNetV2_x0_5 name: MobileNetV2_x0_5
class_num: 1000
# 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: 1000
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: MobileNetV2_x0_75 name: MobileNetV2_x0_75
class_num: 1000
# 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: 1000
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: MobileNetV2_x1_5 name: MobileNetV2_x1_5
class_num: 1000
# 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: 1000
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: MobileNetV2_x2_0 name: MobileNetV2_x2_0
class_num: 1000
# 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: 1000
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_x0_35 name: MobileNetV3_large_x0_35
class_num: 1000
# 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: 1000
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_x0_5 name: MobileNetV3_large_x0_5
class_num: 1000
# 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: 1000
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_x0_75 name: MobileNetV3_large_x0_75
class_num: 1000
# 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: 1000
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: 1000
# 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: 1000
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_25 name: MobileNetV3_large_x1_25
class_num: 1000
# 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: 1000
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_small_x0_35 name: MobileNetV3_small_x0_35
class_num: 1000
# 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: 1000
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_small_x0_5 name: MobileNetV3_small_x0_5
class_num: 1000
# 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: 1000
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_small_x0_75 name: MobileNetV3_small_x0_75
class_num: 1000
# 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: 1000
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_small_x1_0 name: MobileNetV3_small_x1_0
class_num: 1000
# 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: 1000
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_small_x1_25 name: MobileNetV3_small_x1_25
class_num: 1000
# 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: 1000
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: RedNet101 name: RedNet101
class_num: 1000
# 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: 1000
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: RedNet152 name: RedNet152
class_num: 1000
# 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: 1000
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: RedNet26 name: RedNet26
class_num: 1000
# 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: 1000
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: RedNet38 name: RedNet38
class_num: 1000
# 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: 1000
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: RedNet50 name: RedNet50
class_num: 1000
# 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: 1000
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: Res2Net101_vd_26w_4s name: Res2Net101_vd_26w_4s
class_num: 1000
# 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: 1000
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: Res2Net200_vd_26w_4s name: Res2Net200_vd_26w_4s
class_num: 1000
# 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: 1000
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: Res2Net50_14w_8s name: Res2Net50_14w_8s
class_num: 1000
# 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: 1000
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: Res2Net50_26w_4s name: Res2Net50_26w_4s
class_num: 1000
# 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: 1000
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: Res2Net50_vd_26w_4s name: Res2Net50_vd_26w_4s
class_num: 1000
# 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: 1000
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: ResNeSt101 name: ResNeSt101
class_num: 1000
# 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: 1000
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: ResNeSt50 name: ResNeSt50
class_num: 1000
# 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: 1000
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: ResNeSt50_fast_1s1x64d name: ResNeSt50_fast_1s1x64d
class_num: 1000
# 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: 1000
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: ResNeXt101_32x4d name: ResNeXt101_32x4d
class_num: 1000
# 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: 1000
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: ResNeXt101_64x4d name: ResNeXt101_64x4d
class_num: 1000
# 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: 1000
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: ResNeXt101_vd_32x4d name: ResNeXt101_vd_32x4d
class_num: 1000
# 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: 1000
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: ResNeXt101_vd_64x4d name: ResNeXt101_vd_64x4d
class_num: 1000
# 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: 1000
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: ResNeXt152_32x4d name: ResNeXt152_32x4d
class_num: 1000
# 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: 1000
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: ResNeXt152_64x4d name: ResNeXt152_64x4d
class_num: 1000
# 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: 1000
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: ResNeXt152_vd_32x4d name: ResNeXt152_vd_32x4d
class_num: 1000
# 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: 1000
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: ResNeXt152_vd_64x4d name: ResNeXt152_vd_64x4d
class_num: 1000
# 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: 1000
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: ResNeXt50_32x4d name: ResNeXt50_32x4d
class_num: 1000
# 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: 1000
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: ResNeXt50_64x4d name: ResNeXt50_64x4d
class_num: 1000
# 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: 1000
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: ResNeXt50_vd_32x4d name: ResNeXt50_vd_32x4d
class_num: 1000
# 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: 1000
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: ResNeXt50_vd_64x4d name: ResNeXt50_vd_64x4d
class_num: 1000
# 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: 1000
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:
to_static: False to_static: False
# model architecture # model architecture
class_num: 1000
Arch: Arch:
name: ResNet101 name: ResNet101
......
...@@ -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: 1000
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: ResNet101_vd name: ResNet101_vd
class_num: 1000
# 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: 1000
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:
to_static: False to_static: False
# model architecture # model architecture
class_num: 1000
Arch: Arch:
name: ResNet152 name: ResNet152
......
...@@ -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: 1000
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: ResNet152_vd name: ResNet152_vd
class_num: 1000
# 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: 1000
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: ResNet18 name: ResNet18
class_num: 1000
# 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: 1000
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: ResNet18_vd name: ResNet18_vd
class_num: 1000
# 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: 1000
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: ResNet200_vd name: ResNet200_vd
class_num: 1000
# 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: 1000
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: ResNet34 name: ResNet34
class_num: 1000
# 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: 1000
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: ResNet34_vd name: ResNet34_vd
class_num: 1000
# 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: 1000
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:
to_static: False to_static: False
# model architecture # model architecture
class_num: 1000
Arch: Arch:
name: ResNet50 name: ResNet50
......
...@@ -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: 1000
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: 1000
# 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: 1000
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: SENet154_vd name: SENet154_vd
class_num: 1000
# 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: 1000
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: SE_ResNeXt101_32x4d name: SE_ResNeXt101_32x4d
class_num: 1000
# 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: 1000
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: SE_ResNeXt50_32x4d name: SE_ResNeXt50_32x4d
class_num: 1000
# 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: 1000
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: SE_ResNeXt50_vd_32x4d name: SE_ResNeXt50_vd_32x4d
class_num: 1000
# 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: 1000
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: SE_ResNet18_vd name: SE_ResNet18_vd
class_num: 1000
# 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: 1000
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: SE_ResNet34_vd name: SE_ResNet34_vd
class_num: 1000
# 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: 1000
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: SE_ResNet50_vd name: SE_ResNet50_vd
class_num: 1000
# 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: 1000
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: 1000
# 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: 1000
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_33 name: ShuffleNetV2_x0_33
class_num: 1000
# 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: 1000
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_5 name: ShuffleNetV2_x0_5
class_num: 1000
# 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: 1000
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_x1_0 name: ShuffleNetV2_x1_0
class_num: 1000
# 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: 1000
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_x1_5 name: ShuffleNetV2_x1_5
class_num: 1000
# 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: 1000
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_x2_0 name: ShuffleNetV2_x2_0
class_num: 1000
# 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: 1000
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: SqueezeNet1_0 name: SqueezeNet1_0
class_num: 1000
# 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: 1000
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: SqueezeNet1_1 name: SqueezeNet1_1
class_num: 1000
# 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: 1000
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:
to_static: False to_static: False
# model architecture # model architecture
class_num: 1000
Arch: Arch:
name: SwinTransformer_base_patch4_window12_384 name: SwinTransformer_base_patch4_window12_384
......
...@@ -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: 1000
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:
to_static: False to_static: False
# model architecture # model architecture
class_num: 1000
Arch: Arch:
name: SwinTransformer_base_patch4_window7_224 name: SwinTransformer_base_patch4_window7_224
......
...@@ -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: 1000
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:
to_static: False to_static: False
# model architecture # model architecture
class_num: 1000
Arch: Arch:
name: SwinTransformer_large_patch4_window12_384 name: SwinTransformer_large_patch4_window12_384
......
...@@ -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: 1000
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:
to_static: False to_static: False
# model architecture # model architecture
class_num: 1000
Arch: Arch:
name: SwinTransformer_large_patch4_window7_224 name: SwinTransformer_large_patch4_window7_224
......
...@@ -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: 1000
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:
to_static: False to_static: False
# model architecture # model architecture
class_num: 1000
Arch: Arch:
name: SwinTransformer_small_patch4_window7_224 name: SwinTransformer_small_patch4_window7_224
......
...@@ -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: 1000
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:
to_static: False to_static: False
# model architecture # model architecture
class_num: 1000
Arch: Arch:
name: SwinTransformer_tiny_patch4_window7_224 name: SwinTransformer_tiny_patch4_window7_224
......
...@@ -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: 1000
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: TNT_small name: TNT_small
class_num: 1000
# 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: 1000
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: VGG11 name: VGG11
class_num: 1000
# 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: 1000
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: VGG13 name: VGG13
class_num: 1000
# 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: 1000
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: VGG16 name: VGG16
class_num: 1000
# 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: 1000
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: VGG19 name: VGG19
class_num: 1000
# 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: 1000
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: ViT_base_patch16_224 name: ViT_base_patch16_224
class_num: 1000
# 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: 1000
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: ViT_base_patch16_224 name: ViT_base_patch16_224
class_num: 1000
# 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: 1000
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: ViT_base_patch32_384 name: ViT_base_patch32_384
class_num: 1000
# 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: 1000
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: ViT_huge_patch16_224 name: ViT_huge_patch16_224
class_num: 1000
# 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: 1000
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: ViT_huge_patch32_384 name: ViT_huge_patch32_384
class_num: 1000
# 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: 1000
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: ViT_large_patch16_224 name: ViT_large_patch16_224
class_num: 1000
# 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: 1000
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: ViT_large_patch16_384 name: ViT_large_patch16_384
class_num: 1000
# 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: 1000
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: ViT_large_patch32_384 name: ViT_large_patch32_384
class_num: 1000
# 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: 1000
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: ViT_small_patch16_224 name: ViT_small_patch16_224
class_num: 1000
# 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: 1000
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: Xception41 name: Xception41
class_num: 1000
# 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: 1000
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: Xception41_deeplab name: Xception41_deeplab
class_num: 1000
# 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: 1000
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: Xception41_deeplab name: Xception41_deeplab
class_num: 1000
# 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: 1000
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: Xception65_deeplab name: Xception65_deeplab
class_num: 1000
# 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: 1000
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: Xception71 name: Xception71
class_num: 1000
# 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: 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:
......
...@@ -67,8 +67,6 @@ class Trainer(object): ...@@ -67,8 +67,6 @@ class Trainer(object):
dist.init_parallel_env() dist.init_parallel_env()
if "Head" in self.config["Arch"]: if "Head" in self.config["Arch"]:
self.config["Arch"]["Head"]["class_num"] = self.config["Global"][
"class_num"]
self.is_rec = True self.is_rec = True
else: else:
self.is_rec = False self.is_rec = False
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册