提交 830852e7 编写于 作者: weixin_46524038's avatar weixin_46524038 提交者: cuicheng01

modify configs

上级 e238686e
......@@ -18,7 +18,7 @@ Global:
Arch:
name: RegNetX_12GF
class_num: 1000
# loss function config for traing/eval process
Loss:
Train:
......@@ -127,4 +127,4 @@ Metric:
topk: [1, 5]
Eval:
- TopkAcc:
topk: [1, 5]
\ No newline at end of file
topk: [1, 5]
......@@ -18,7 +18,7 @@ Global:
Arch:
name: RegNetX_1600MF
class_num: 1000
# loss function config for traing/eval process
Loss:
Train:
......@@ -127,4 +127,4 @@ Metric:
topk: [1, 5]
Eval:
- TopkAcc:
topk: [1, 5]
\ No newline at end of file
topk: [1, 5]
......@@ -18,7 +18,7 @@ Global:
Arch:
name: RegNetX_16GF
class_num: 1000
# loss function config for traing/eval process
Loss:
Train:
......@@ -127,4 +127,4 @@ Metric:
topk: [1, 5]
Eval:
- TopkAcc:
topk: [1, 5]
\ No newline at end of file
topk: [1, 5]
......@@ -92,7 +92,7 @@ DataLoader:
order: ''
sampler:
name: DistributedBatchSampler
batch_size: 128
batch_size: 64
drop_last: False
shuffle: False
loader:
......
......@@ -18,7 +18,7 @@ Global:
Arch:
name: RepVGG_B3
class_num: 1000
# loss function config for traing/eval process
Loss:
Train:
......@@ -134,4 +134,4 @@ Metric:
topk: [1, 5]
Eval:
- TopkAcc:
topk: [1, 5]
\ No newline at end of file
topk: [1, 5]
......@@ -134,4 +134,4 @@ Metric:
topk: [1, 5]
Eval:
- TopkAcc:
topk: [1, 5]
\ No newline at end of file
topk: [1, 5]
......@@ -18,7 +18,7 @@ Global:
Arch:
name: RepVGG_D2se
class_num: 1000
# loss function config for traing/eval process
Loss:
Train:
......@@ -134,4 +134,4 @@ Metric:
topk: [1, 5]
Eval:
- TopkAcc:
topk: [1, 5]
\ No newline at end of file
topk: [1, 5]
......@@ -128,4 +128,4 @@ Metric:
Train:
Eval:
- TopkAcc:
topk: [1, 5]
\ No newline at end of file
topk: [1, 5]
......@@ -11,7 +11,7 @@ Global:
print_batch_step: 10
use_visualdl: False
# used for static mode and model export
image_shape: [3, 256, 256]
image_shape: [3, 320, 320]
save_inference_dir: ./inference
# model architecture
......@@ -53,7 +53,7 @@ DataLoader:
to_rgb: True
channel_first: False
- RandCropImage:
size: 256
size: 320
- RandFlipImage:
flip_code: 1
- AutoAugment:
......@@ -85,9 +85,9 @@ DataLoader:
to_rgb: True
channel_first: False
- ResizeImage:
resize_short: 288
resize_short: 320
- CropImage:
size: 256
size: 320
- NormalizeImage:
scale: 1.0/255.0
mean: [0.485, 0.456, 0.406]
......@@ -110,9 +110,9 @@ Infer:
to_rgb: True
channel_first: False
- ResizeImage:
resize_short: 288
resize_short: 320
- CropImage:
size: 256
size: 320
- NormalizeImage:
scale: 1.0/255.0
mean: [0.485, 0.456, 0.406]
......@@ -128,4 +128,4 @@ Metric:
Train:
Eval:
- TopkAcc:
topk: [1, 5]
\ No newline at end of file
topk: [1, 5]
......@@ -11,7 +11,7 @@ Global:
print_batch_step: 10
use_visualdl: False
# used for static mode and model export
image_shape: [3, 256, 256]
image_shape: [3, 416, 416]
save_inference_dir: ./inference
# model architecture
......@@ -53,7 +53,7 @@ DataLoader:
to_rgb: True
channel_first: False
- RandCropImage:
size: 256
size: 416
- RandFlipImage:
flip_code: 1
- AutoAugment:
......@@ -85,9 +85,9 @@ DataLoader:
to_rgb: True
channel_first: False
- ResizeImage:
resize_short: 288
resize_short: 416
- CropImage:
size: 256
size: 416
- NormalizeImage:
scale: 1.0/255.0
mean: [0.485, 0.456, 0.406]
......@@ -110,9 +110,9 @@ Infer:
to_rgb: True
channel_first: False
- ResizeImage:
resize_short: 288
resize_short: 416
- CropImage:
size: 256
size: 416
- NormalizeImage:
scale: 1.0/255.0
mean: [0.485, 0.456, 0.406]
......@@ -128,4 +128,4 @@ Metric:
Train:
Eval:
- TopkAcc:
topk: [1, 5]
\ No newline at end of file
topk: [1, 5]
......@@ -18,7 +18,7 @@ Global:
Arch:
name: TNT_base
class_num: 1000
# loss function config for traing/eval process
Loss:
Train:
......@@ -131,4 +131,4 @@ Metric:
topk: [1, 5]
Eval:
- TopkAcc:
topk: [1, 5]
\ No newline at end of file
topk: [1, 5]
......@@ -131,4 +131,4 @@ Metric:
topk: [1, 5]
Eval:
- TopkAcc:
topk: [1, 5]
\ No newline at end of file
topk: [1, 5]
......@@ -22,7 +22,7 @@ Arch:
class_num: 1000
drop_path_rate: 0.1
drop_rate: 0.0
# loss function config for traing/eval process
Loss:
Train:
......@@ -155,4 +155,4 @@ Infer:
Metric:
Eval:
- TopkAcc:
topk: [1, 5]
\ No newline at end of file
topk: [1, 5]
......@@ -22,7 +22,7 @@ Arch:
class_num: 1000
drop_path_rate: 0.1
drop_rate: 0.0
# loss function config for traing/eval process
Loss:
Train:
......@@ -155,4 +155,4 @@ Infer:
Metric:
Eval:
- TopkAcc:
topk: [1, 5]
\ No newline at end of file
topk: [1, 5]
......@@ -22,7 +22,7 @@ Arch:
class_num: 1000
drop_path_rate: 0.1
drop_rate: 0.0
# loss function config for traing/eval process
Loss:
Train:
......@@ -155,4 +155,4 @@ Infer:
Metric:
Eval:
- TopkAcc:
topk: [1, 5]
\ No newline at end of file
topk: [1, 5]
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册