fix config files (#344)
parent
8d046546c8
commit
806675d226
|
@ -13,6 +13,9 @@ epochs: 200
|
|||
topk: 5
|
||||
image_shape: [3, 256, 256]
|
||||
|
||||
use_mix: True
|
||||
ls_epsilon: 0.1
|
||||
|
||||
LEARNING_RATE:
|
||||
function: 'Cosine'
|
||||
params:
|
||||
|
@ -47,6 +50,9 @@ TRAIN:
|
|||
std: [0.229, 0.224, 0.225]
|
||||
order: ''
|
||||
- ToCHWImage:
|
||||
mix:
|
||||
- MixupOperator:
|
||||
alpha: 0.2
|
||||
|
||||
VALID:
|
||||
batch_size: 64
|
||||
|
|
|
@ -24,7 +24,7 @@ OPTIMIZER:
|
|||
momentum: 0.9
|
||||
regularizer:
|
||||
function: 'L2'
|
||||
factor: 0.0004
|
||||
factor: 0.0002
|
||||
|
||||
TRAIN:
|
||||
batch_size: 512
|
||||
|
|
Loading…
Reference in New Issue