diff --git a/ppcls/configs/ImageNet/PVTV2/PVT_V2_B0.yaml b/ppcls/configs/ImageNet/PVTV2/PVT_V2_B0.yaml index 2c60eccad96b311b9f9c31a1a73baa4d63b9b21d..27684e9cdfb24ca975f32b1f1179b664c34e8f35 100644 --- a/ppcls/configs/ImageNet/PVTV2/PVT_V2_B0.yaml +++ b/ppcls/configs/ImageNet/PVTV2/PVT_V2_B0.yaml @@ -45,9 +45,9 @@ Optimizer: lr: name: Cosine learning_rate: 5e-4 - eta_min: 1e-5 + eta_min: 5e-6 warmup_epoch: 20 - warmup_start_lr: 1e-6 + warmup_start_lr: 5e-7 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/PVTV2/PVT_V2_B1.yaml b/ppcls/configs/ImageNet/PVTV2/PVT_V2_B1.yaml index 04fcf0c53a1f9bc71dbb05586a5c30b1d9ce1ee8..2e3e176c8123aba7324546bb69a8af1971a5b92f 100644 --- a/ppcls/configs/ImageNet/PVTV2/PVT_V2_B1.yaml +++ b/ppcls/configs/ImageNet/PVTV2/PVT_V2_B1.yaml @@ -45,9 +45,9 @@ Optimizer: lr: name: Cosine learning_rate: 5e-4 - eta_min: 1e-5 + eta_min: 5e-6 warmup_epoch: 20 - warmup_start_lr: 1e-6 + warmup_start_lr: 5e-7 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/PVTV2/PVT_V2_B2.yaml b/ppcls/configs/ImageNet/PVTV2/PVT_V2_B2.yaml index 8c7bc832cbce41d6a23467df7cdeb1ff6ae12968..ce81bccaa3400e7c0237a5c7854ec8e893c7ea59 100644 --- a/ppcls/configs/ImageNet/PVTV2/PVT_V2_B2.yaml +++ b/ppcls/configs/ImageNet/PVTV2/PVT_V2_B2.yaml @@ -45,9 +45,9 @@ Optimizer: lr: name: Cosine learning_rate: 5e-4 - eta_min: 1e-5 + eta_min: 5e-6 warmup_epoch: 20 - warmup_start_lr: 1e-6 + warmup_start_lr: 5e-7 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/PVTV2/PVT_V2_B2_Linear.yaml b/ppcls/configs/ImageNet/PVTV2/PVT_V2_B2_Linear.yaml index ce84d93059a9198e5abb5523efb8c33bccd17186..5844dccc4a9c61355f97c6e4749968f5e7887b99 100644 --- a/ppcls/configs/ImageNet/PVTV2/PVT_V2_B2_Linear.yaml +++ b/ppcls/configs/ImageNet/PVTV2/PVT_V2_B2_Linear.yaml @@ -45,9 +45,9 @@ Optimizer: lr: name: Cosine learning_rate: 5e-4 - eta_min: 1e-5 + eta_min: 5e-6 warmup_epoch: 20 - warmup_start_lr: 1e-6 + warmup_start_lr: 5e-7 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/PVTV2/PVT_V2_B3.yaml b/ppcls/configs/ImageNet/PVTV2/PVT_V2_B3.yaml index 4efcdc89f5bf86e81a711e64a2e1ea98a20fad51..7b26cd350c65409159264ee2ec7b92107cbb48c6 100644 --- a/ppcls/configs/ImageNet/PVTV2/PVT_V2_B3.yaml +++ b/ppcls/configs/ImageNet/PVTV2/PVT_V2_B3.yaml @@ -46,9 +46,9 @@ Optimizer: lr: name: Cosine learning_rate: 5e-4 - eta_min: 1e-5 + eta_min: 5e-6 warmup_epoch: 20 - warmup_start_lr: 1e-6 + warmup_start_lr: 5e-7 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/PVTV2/PVT_V2_B4.yaml b/ppcls/configs/ImageNet/PVTV2/PVT_V2_B4.yaml index 0f56427f852ab954ae8db92c56e09dd8123c4738..30b63a201da246d61870375ab7fc5995883a25ae 100644 --- a/ppcls/configs/ImageNet/PVTV2/PVT_V2_B4.yaml +++ b/ppcls/configs/ImageNet/PVTV2/PVT_V2_B4.yaml @@ -46,9 +46,9 @@ Optimizer: lr: name: Cosine learning_rate: 5e-4 - eta_min: 1e-5 + eta_min: 5e-6 warmup_epoch: 20 - warmup_start_lr: 1e-6 + warmup_start_lr: 5e-7 # data loader for train and eval diff --git a/ppcls/configs/ImageNet/PVTV2/PVT_V2_B5.yaml b/ppcls/configs/ImageNet/PVTV2/PVT_V2_B5.yaml index 53ded6495e5d67456292faeb9e11be0032d56b1b..0992d5805d3ba99835aa50acad2c1e1cb4c0b6aa 100644 --- a/ppcls/configs/ImageNet/PVTV2/PVT_V2_B5.yaml +++ b/ppcls/configs/ImageNet/PVTV2/PVT_V2_B5.yaml @@ -46,9 +46,9 @@ Optimizer: lr: name: Cosine learning_rate: 5e-4 - eta_min: 1e-5 + eta_min: 5e-6 warmup_epoch: 20 - warmup_start_lr: 1e-6 + warmup_start_lr: 5e-7 # data loader for train and eval