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

Merge pull request #988 from littletomatodonkey/dev/fix_dim_cfg

fix yaml
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: "./output/"
device: "gpu"
class_num: 5013
save_interval: 1
eval_mode: "retrieval"
eval_during_train: True
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: AlexNet
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DLA102
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DLA102x
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DLA102x2
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DLA169
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DLA34
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DLA46_c
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DLA46x_c
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DLA60
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DLA60x
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DLA60x_c
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DPN107
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DPN131
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DPN68
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DPN92
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DPN98
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DarkNet53
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNet50
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNet50
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNet50
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNet50
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNet50
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNet50
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNet50
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNet50
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNet50
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DeiT_base_distilled_patch16_224
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DeiT_base_distilled_patch16_384
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DeiT_base_patch16_224
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DeiT_base_patch16_384
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DeiT_small_distilled_patch16_224
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DeiT_small_patch16_224
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DeiT_tiny_distilled_patch16_224
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DeiT_tiny_patch16_224
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DenseNet121
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DenseNet161
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DenseNet169
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DenseNet201
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: DenseNet264
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: "./output/"
device: "gpu"
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: "DistillationModel"
class_num: 1000
# if not null, its lengths should be same as models
pretrained_list:
# if not null, its lengths should be same as models
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: EfficientNetB0
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: GhostNet_x0_5
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: GhostNet_x1_0
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: GhostNet_x1_3
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: HRNet_W18_C
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: HRNet_W30_C
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: HRNet_W32_C
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: HRNet_W40_C
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: HRNet_W44_C
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: HRNet_W48_C
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: HRNet_W64_C
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: HarDNet39_ds
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: HarDNet68
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: HarDNet68_ds
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: HarDNet85
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: GoogLeNet
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: InceptionV3
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: InceptionV4
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: LeViT_128
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: LeViT_128S
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: LeViT_192
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: LeViT_256
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: LeViT_384
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
to_static: False
# model architecture
class_num: 1000
Arch:
name: MobileNetV1
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: MobileNetV1_x0_25
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: MobileNetV1_x0_5
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: MobileNetV1_x0_75
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
to_static: False
# model architecture
class_num: 1000
Arch:
name: MobileNetV2
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: MobileNetV2_x0_25
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: MobileNetV2_x0_5
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: MobileNetV2_x0_75
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: MobileNetV2_x1_5
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: MobileNetV2_x2_0
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: MobileNetV3_large_x0_35
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: MobileNetV3_large_x0_5
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: MobileNetV3_large_x0_75
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: MobileNetV3_large_x1_0
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: MobileNetV3_large_x1_25
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: MobileNetV3_small_x0_35
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: MobileNetV3_small_x0_5
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: MobileNetV3_small_x0_75
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: MobileNetV3_small_x1_0
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: MobileNetV3_small_x1_25
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: RedNet101
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: RedNet152
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: RedNet26
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: RedNet38
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: RedNet50
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: Res2Net101_vd_26w_4s
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: Res2Net200_vd_26w_4s
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: Res2Net50_14w_8s
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: Res2Net50_26w_4s
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: Res2Net50_vd_26w_4s
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNeSt101
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNeSt50
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNeSt50_fast_1s1x64d
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNeXt101_32x4d
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNeXt101_64x4d
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNeXt101_vd_32x4d
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNeXt101_vd_64x4d
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNeXt152_32x4d
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNeXt152_64x4d
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNeXt152_vd_32x4d
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNeXt152_vd_64x4d
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNeXt50_32x4d
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNeXt50_64x4d
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNeXt50_vd_32x4d
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNeXt50_vd_64x4d
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
to_static: False
# model architecture
class_num: 1000
Arch:
name: ResNet101
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNet101_vd
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
to_static: False
# model architecture
class_num: 1000
Arch:
name: ResNet152
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNet152_vd
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNet18
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNet18_vd
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNet200_vd
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNet34
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNet34_vd
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
to_static: False
# model architecture
class_num: 1000
Arch:
name: ResNet50
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNet50_vd
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: SENet154_vd
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: SE_ResNeXt101_32x4d
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: SE_ResNeXt50_32x4d
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: SE_ResNeXt50_vd_32x4d
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: SE_ResNet18_vd
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: SE_ResNet34_vd
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: SE_ResNet50_vd
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ShuffleNetV2_x0_25
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ShuffleNetV2_x0_33
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ShuffleNetV2_x0_5
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ShuffleNetV2_x1_0
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ShuffleNetV2_x1_5
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ShuffleNetV2_x2_0
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: SqueezeNet1_0
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: SqueezeNet1_1
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
to_static: False
# model architecture
class_num: 1000
Arch:
name: SwinTransformer_base_patch4_window12_384
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
to_static: False
# model architecture
class_num: 1000
Arch:
name: SwinTransformer_base_patch4_window7_224
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
to_static: False
# model architecture
class_num: 1000
Arch:
name: SwinTransformer_large_patch4_window12_384
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
to_static: False
# model architecture
class_num: 1000
Arch:
name: SwinTransformer_large_patch4_window7_224
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
to_static: False
# model architecture
class_num: 1000
Arch:
name: SwinTransformer_small_patch4_window7_224
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
to_static: False
# model architecture
class_num: 1000
Arch:
name: SwinTransformer_tiny_patch4_window7_224
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: TNT_small
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: VGG11
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: VGG13
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: VGG16
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: VGG19
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ViT_base_patch16_224
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ViT_base_patch16_224
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ViT_base_patch32_384
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ViT_huge_patch16_224
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ViT_huge_patch32_384
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ViT_large_patch16_224
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ViT_large_patch16_384
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ViT_large_patch32_384
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ViT_small_patch16_224
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: Xception41
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: Xception41_deeplab
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: Xception41_deeplab
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: Xception65_deeplab
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 1000
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: Xception71
class_num: 1000
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: "./output/"
device: "gpu"
class_num: 3000
save_interval: 1
eval_during_train: True
eval_interval: 1
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 50030
save_interval: 1
eval_during_train: True
eval_interval: 10
......
......@@ -6,7 +6,6 @@ Global:
pretrained_model: product_ResNet50_vd_Aliproduct_v1.0_pretrained
output_dir: ./output/
device: gpu
class_num: 3997
save_interval: 10
eval_during_train: True
eval_interval: 10
......
......@@ -6,7 +6,6 @@ Global:
pretrained_model: product_ResNet50_vd_Aliproduct_v1.0_pretrained
output_dir: ./output/
device: gpu
class_num: 11319
save_interval: 10
eval_during_train: True
eval_interval: 10
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: "./output/"
device: "gpu"
class_num: 431
save_interval: 1
eval_during_train: True
eval_interval: 1
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: "./output/"
device: "gpu"
class_num: 30671
save_interval: 1
eval_during_train: True
eval_interval: 1
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 101
save_interval: 5
eval_during_train: True
eval_interval: 1
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 102
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: MobileNetV3_large_x1_0
class_num: 102
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: cpu
class_num: 102
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ShuffleNetV2_x0_25
class_num: 102
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 100
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: MobileNetV3_large_x1_0
class_num: 100
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: "./output/"
device: "gpu"
class_num: 100
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -27,9 +26,11 @@ Arch:
models:
- Teacher:
name: ResNet50_vd
class_num: 100
pretrained: "./pretrained/best_model"
- Student:
name: MobileNetV3_large_x1_0
class_num: 100
pretrained: True
infer_model_name: "Student"
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 100
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNet50_vd
class_num: 100
# loss function config for traing/eval process
Loss:
......
......@@ -4,7 +4,6 @@ Global:
pretrained_model: null
output_dir: ./output/
device: gpu
class_num: 100
save_interval: 1
eval_during_train: True
eval_interval: 1
......@@ -18,6 +17,7 @@ Global:
# model architecture
Arch:
name: ResNet50_vd
class_num: 100
# loss function config for traing/eval process
Loss:
......
......@@ -67,8 +67,6 @@ class Trainer(object):
dist.init_parallel_env()
if "Head" in self.config["Arch"]:
self.config["Arch"]["Head"]["class_num"] = self.config["Global"][
"class_num"]
self.is_rec = True
else:
self.is_rec = False
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册