diff --git a/configs/vitdet/_base_/optimizer_base_30e.yml b/configs/vitdet/_base_/optimizer_base_30e.yml deleted file mode 100644 index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..0000000000000000000000000000000000000000 diff --git a/configs/vitdet/_base_/optimizer_base_36e.yml b/configs/vitdet/_base_/optimizer_base_36e.yml new file mode 100644 index 0000000000000000000000000000000000000000..83b8708d046ea2ce57345ba543ed39453389f45d --- /dev/null +++ b/configs/vitdet/_base_/optimizer_base_36e.yml @@ -0,0 +1,20 @@ + +epoch: 36 + +LearningRate: + base_lr: 0.0001 + schedulers: + - !CosineDecay + max_epochs: 36 + min_lr_ratio: 0.1 # 0.1 + - !LinearWarmup + start_factor: 0.001 + epochs: 1 + + +OptimizerBuilder: + clip_grad_by_norm: 0.1 + regularizer: false + optimizer: + type: AdamW + weight_decay: 0.0001 diff --git a/configs/vitdet/ppyoloe_vit_base_csppan_cae_30e_coco.yml b/configs/vitdet/ppyoloe_vit_base_csppan_cae_36e_coco.yml similarity index 93% rename from configs/vitdet/ppyoloe_vit_base_csppan_cae_30e_coco.yml rename to configs/vitdet/ppyoloe_vit_base_csppan_cae_36e_coco.yml index d446d5cc9152d9de4a7fb36fb233680008eb6d03..556f4b49d7a2a1c7c6928b4e917881b175b01384 100644 --- a/configs/vitdet/ppyoloe_vit_base_csppan_cae_30e_coco.yml +++ b/configs/vitdet/ppyoloe_vit_base_csppan_cae_36e_coco.yml @@ -3,10 +3,10 @@ _BASE_: [ '../datasets/coco_detection.yml', '../runtime.yml', './_base_/ppyoloe_reader.yml', - './_base_/optimizer_base_30e.yml' + './_base_/optimizer_base_36e.yml' ] -weights: output/ppyoloe_vit_base_csppan_cae_30e_coco/model_final +weights: output/ppyoloe_vit_base_csppan_cae_36e_coco/model_final snapshot_epoch: 2