Skip to content

Commit 876d485

Browse files
committed
Update learning_rate.py
1 parent ddaeaa6 commit 876d485

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

deepmd/dpmodel/utils/learning_rate.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ def __init__(
154154
discrete updates every `decay_steps` steps (akin to treating each
155155
update as an "epoch").
156156
"""
157-
self.base_lr = float(start_lr)
157+
self.start_lr = float(start_lr)
158158
self.start_factor = float(start_factor)
159159
self.end_factor = float(end_factor)
160160
self.stop_steps = int(stop_steps)
@@ -200,4 +200,4 @@ def value(self, step: int) -> np.float64:
200200
else:
201201
factor = min(factor, self.end_factor)
202202

203-
return np.float64(self.base_lr * factor)
203+
return np.float64(self.start_lr * factor)

0 commit comments

Comments
 (0)