rm pdmodel (#3287)

* rm pdmodel * fix params
PaddlePaddle · Nov 5, 2024 · 1cde9bc · 1cde9bc
1 parent fffe2e9
commit 1cde9bc
Show file tree

Hide file tree

Showing 11 changed files with 10 additions and 10 deletions.
diff --git a/MobileNetV4_hybrid_medium_pretrained.pdparams b/MobileNetV4_hybrid_medium_pretrained.pdparams
diff --git a/ppcls/configs/ImageNet/FasterNet/FasterNet_L.yaml b/ppcls/configs/ImageNet/FasterNet/FasterNet_L.yaml
@@ -94,7 +94,7 @@ DataLoader:
               prob: 0.5
     sampler:
       name: DistributedBatchSampler
-      batch_size: 50
+      batch_size: 256
       drop_last: False
       shuffle: True
     loader:

diff --git a/ppcls/configs/ImageNet/FasterNet/FasterNet_M.yaml b/ppcls/configs/ImageNet/FasterNet/FasterNet_M.yaml
@@ -94,7 +94,7 @@ DataLoader:
               prob: 0.5
     sampler:
       name: DistributedBatchSampler
-      batch_size: 50
+      batch_size: 256
       drop_last: False
       shuffle: True
     loader:

diff --git a/ppcls/configs/ImageNet/FasterNet/FasterNet_S.yaml b/ppcls/configs/ImageNet/FasterNet/FasterNet_S.yaml
@@ -94,7 +94,7 @@ DataLoader:
               prob: 0.5
     sampler:
       name: DistributedBatchSampler
-      batch_size: 50
+      batch_size: 512
       drop_last: False
       shuffle: True
     loader:

diff --git a/ppcls/configs/ImageNet/FasterNet/FasterNet_T0.yaml b/ppcls/configs/ImageNet/FasterNet/FasterNet_T0.yaml
@@ -94,7 +94,7 @@ DataLoader:
               prob: 0.5
     sampler:
       name: DistributedBatchSampler
-      batch_size: 50
+      batch_size: 512
       drop_last: False
       shuffle: True
     loader:

diff --git a/ppcls/configs/ImageNet/FasterNet/FasterNet_T1.yaml b/ppcls/configs/ImageNet/FasterNet/FasterNet_T1.yaml
@@ -94,7 +94,7 @@ DataLoader:
               prob: 0.5
     sampler:
       name: DistributedBatchSampler
-      batch_size: 50
+      batch_size: 512
       drop_last: False
       shuffle: True
     loader:

diff --git a/ppcls/configs/ImageNet/FasterNet/FasterNet_T2.yaml b/ppcls/configs/ImageNet/FasterNet/FasterNet_T2.yaml
@@ -94,7 +94,7 @@ DataLoader:
               prob: 0.5
     sampler:
       name: DistributedBatchSampler
-      batch_size: 50
+      batch_size: 512
       drop_last: False
       shuffle: True
     loader:

diff --git a/ppcls/configs/ImageNet/MobileNetV4/MobileNetV4_conv_large.yaml b/ppcls/configs/ImageNet/MobileNetV4/MobileNetV4_conv_large.yaml
@@ -57,7 +57,7 @@ Optimizer:
   lr:
     # for 8 cards
     name: Cosine
-    learning_rate: 0.002 ##null
+    learning_rate: 0.003 ##null
     eta_min: 1.0e-06
     warmup_epoch: 20
     warmup_start_lr: 0

diff --git a/ppcls/configs/ImageNet/MobileNetV4/MobileNetV4_conv_small.yaml b/ppcls/configs/ImageNet/MobileNetV4/MobileNetV4_conv_small.yaml
@@ -111,7 +111,7 @@ DataLoader:
               prob: 0.5
     sampler:
       name: DistributedBatchSampler
-      batch_size: 1024
+      batch_size: 512
       drop_last: False
       shuffle: True
     loader:

diff --git a/ppcls/configs/ImageNet/MobileNetV4/MobileNetV4_hybrid_large.yaml b/ppcls/configs/ImageNet/MobileNetV4/MobileNetV4_hybrid_large.yaml
@@ -30,7 +30,7 @@ AMP:
 Arch:
   name: MobileNetV4_hybrid_large
   drop_rate: 0.2
-  drop_path_rate: 0.1
+  drop_path_rate: 0.35
   use_fused_attn: False
   class_num: 1000
 

diff --git a/ppcls/configs/ImageNet/MobileNetV4/MobileNetV4_hybrid_medium.yaml b/ppcls/configs/ImageNet/MobileNetV4/MobileNetV4_hybrid_medium.yaml
@@ -109,7 +109,7 @@ DataLoader:
               prob: 0.5
     sampler:
       name: DistributedBatchSampler
-      batch_size: 64
+      batch_size: 128
       drop_last: False
       shuffle: True
     loader: