diff --git a/ppcls/configs/ImageNet/GhostNet/GhostNet_x0_5.yaml b/ppcls/configs/ImageNet/GhostNet/GhostNet_x0_5.yaml index ab50f8a2410034c7f53aa065f309e082f0a70c48..ecf64c0a1d63322de4f1fe6dca31ce504ef912a3 100644 --- a/ppcls/configs/ImageNet/GhostNet/GhostNet_x0_5.yaml +++ b/ppcls/configs/ImageNet/GhostNet/GhostNet_x0_5.yaml @@ -24,6 +24,7 @@ Loss: Train: - CELoss: weight: 1.0 + epsilon: 0.1 Eval: - CELoss: weight: 1.0 @@ -35,9 +36,10 @@ Optimizer: lr: name: Cosine learning_rate: 0.8 + warmup_epoch: 5 regularizer: name: 'L2' - coeff: 0.0004 + coeff: 0.00004 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/GhostNet/GhostNet_x1_0.yaml b/ppcls/configs/ImageNet/GhostNet/GhostNet_x1_0.yaml index d28748a915545cae954115c33a4e071900fe9d63..613aef80bde35f60c4832cb1763f069a90bdb938 100644 --- a/ppcls/configs/ImageNet/GhostNet/GhostNet_x1_0.yaml +++ b/ppcls/configs/ImageNet/GhostNet/GhostNet_x1_0.yaml @@ -24,6 +24,7 @@ Loss: Train: - CELoss: weight: 1.0 + epsilon: 0.1 Eval: - CELoss: weight: 1.0 @@ -35,9 +36,10 @@ Optimizer: lr: name: Cosine learning_rate: 0.8 + warmup_epoch: 5 regularizer: name: 'L2' - coeff: 0.0004 + coeff: 0.00004 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/GhostNet/GhostNet_x1_3.yaml b/ppcls/configs/ImageNet/GhostNet/GhostNet_x1_3.yaml index b6b03c781f8283577c0d6a78a950ed58ce893563..d1d40e0a700f24ff142c6318339e74c9a8f2aa98 100644 --- a/ppcls/configs/ImageNet/GhostNet/GhostNet_x1_3.yaml +++ b/ppcls/configs/ImageNet/GhostNet/GhostNet_x1_3.yaml @@ -24,6 +24,7 @@ Loss: Train: - CELoss: weight: 1.0 + epsilon: 0.1 Eval: - CELoss: weight: 1.0 @@ -35,9 +36,10 @@ Optimizer: lr: name: Cosine learning_rate: 0.8 + warmup_epoch: 5 regularizer: name: 'L2' - coeff: 0.0004 + coeff: 0.00004 # data loader for train and eval