change the lr step size.

54012e90 · Zhen Wang · 377c6d32 · 54012e90 · 54012e90 · 54012e90
Showing with 3 addition and 3 deletion

PaddleSlim/models/googlenet.py PaddleSlim/models/googlenet.py +1 -1

PaddleSlim/models/mobilenet.py PaddleSlim/models/mobilenet.py +1 -1

PaddleSlim/models/resnet.py PaddleSlim/models/resnet.py +1 -1

未找到文件。
--- a/PaddleSlim/models/googlenet.py
+++ b/PaddleSlim/models/googlenet.py
@@ -14,7 +14,7 @@ train_parameters = {
    "learning_strategy": {
        "name": "piecewise_decay",
        "batch_size": 256,
-        "epochs": [30, 70, 100],
+        "epochs": [10, 16, 30],
        "steps": [0.1, 0.01, 0.001, 0.0001]
    }
 }

--- a/PaddleSlim/models/mobilenet.py
+++ b/PaddleSlim/models/mobilenet.py
@@ -14,7 +14,7 @@ train_parameters = {
    "learning_strategy": {
        "name": "piecewise_decay",
        "batch_size": 256,
-        "epochs": [30, 60, 90],
+        "epochs": [10, 16, 30],
        "steps": [0.1, 0.01, 0.001, 0.0001]
    }
 }

--- a/PaddleSlim/models/resnet.py
+++ b/PaddleSlim/models/resnet.py
@@ -15,7 +15,7 @@ train_parameters = {
    "learning_strategy": {
        "name": "piecewise_decay",
        "batch_size": 256,
-        "epochs": [30, 60, 90],
+        "epochs": [10, 16, 30],
        "steps": [0.1, 0.01, 0.001, 0.0001]
    }
 }