diff --git a/configs/DarkNet/DarkNet53.yaml b/configs/DarkNet/DarkNet53.yaml index 525ba711f5596a3b0681d43a93b38392a9df8c3a..1dfe7301052b59af5d10adb5431aef3862918436 100644 --- a/configs/DarkNet/DarkNet53.yaml +++ b/configs/DarkNet/DarkNet53.yaml @@ -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 diff --git a/configs/VGG/VGG11.yaml b/configs/VGG/VGG11.yaml index 9331f52f932b60fdf9fdf0d032d70f133ed1604a..4f26a162bfbdc3f8db187dc7ed890607d4ae5ed6 100644 --- a/configs/VGG/VGG11.yaml +++ b/configs/VGG/VGG11.yaml @@ -24,7 +24,7 @@ OPTIMIZER: momentum: 0.9 regularizer: function: 'L2' - factor: 0.0004 + factor: 0.0002 TRAIN: batch_size: 512