will-jl944 4 жил өмнө
parent
commit
978c98bfd7

+ 1 - 1
dygraph/paddlex/cv/models/classifier.py

@@ -252,7 +252,7 @@ class BaseClassifier(BaseModel):
                           save_interval_epochs=1,
                           log_interval_steps=10,
                           save_dir='output',
-                          learning_rate=.025,
+                          learning_rate=.000025,
                           warmup_steps=0,
                           warmup_start_lr=0.0,
                           lr_decay_epochs=(30, 60, 90),

+ 1 - 1
dygraph/paddlex/cv/models/detector.py

@@ -255,7 +255,7 @@ class BaseDetector(BaseModel):
                           save_interval_epochs=1,
                           log_interval_steps=10,
                           save_dir='output',
-                          learning_rate=.001,
+                          learning_rate=.00001,
                           warmup_steps=0,
                           warmup_start_lr=0.0,
                           lr_decay_epochs=(216, 243),

+ 1 - 1
dygraph/paddlex/cv/models/segmenter.py

@@ -237,7 +237,7 @@ class BaseSegmenter(BaseModel):
                           save_interval_epochs=1,
                           log_interval_steps=2,
                           save_dir='output',
-                          learning_rate=0.01,
+                          learning_rate=0.0001,
                           lr_decay_power=0.9,
                           early_stop=False,
                           early_stop_patience=5,