We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ddaeaa6 commit 876d485Copy full SHA for 876d485
1 file changed
deepmd/dpmodel/utils/learning_rate.py
@@ -154,7 +154,7 @@ def __init__(
154
discrete updates every `decay_steps` steps (akin to treating each
155
update as an "epoch").
156
"""
157
- self.base_lr = float(start_lr)
+ self.start_lr = float(start_lr)
158
self.start_factor = float(start_factor)
159
self.end_factor = float(end_factor)
160
self.stop_steps = int(stop_steps)
@@ -200,4 +200,4 @@ def value(self, step: int) -> np.float64:
200
else:
201
factor = min(factor, self.end_factor)
202
203
- return np.float64(self.base_lr * factor)
+ return np.float64(self.start_lr * factor)
0 commit comments