未验证 提交 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:
......
此差异已折叠。
此差异已折叠。
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册