fix config files (#344)

pull/348/head
littletomatodonkey 2020-10-27 17:36:44 +08:00 committed by GitHub
parent 8d046546c8
commit 806675d226
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 1 deletions

View File

@ -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

View File

@ -24,7 +24,7 @@ OPTIMIZER:
momentum: 0.9
regularizer:
function: 'L2'
factor: 0.0004
factor: 0.0002
TRAIN:
batch_size: 512