From 3decb94a446e5d9e41c1835d30920ea5c1a454c2 Mon Sep 17 00:00:00 2001 From: cuicheng01 Date: Thu, 12 Aug 2021 08:51:05 +0000 Subject: [PATCH] Update GhostNet configs --- ppcls/configs/ImageNet/GhostNet/GhostNet_x0_5.yaml | 4 +++- ppcls/configs/ImageNet/GhostNet/GhostNet_x1_0.yaml | 4 +++- ppcls/configs/ImageNet/GhostNet/GhostNet_x1_3.yaml | 4 +++- 3 files changed, 9 insertions(+), 3 deletions(-) diff --git a/ppcls/configs/ImageNet/GhostNet/GhostNet_x0_5.yaml b/ppcls/configs/ImageNet/GhostNet/GhostNet_x0_5.yaml index ab50f8a2..ecf64c0a 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 d28748a9..613aef80 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 b6b03c78..d1d40e0a 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 -- GitLab