We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1a1dc59 commit 716a15bCopy full SHA for 716a15b
1 file changed
deepmd/pt/train/training.py
@@ -164,7 +164,7 @@ def __init__(
164
165
# Iteration config
166
self.num_steps = training_params.get("numb_steps")
167
- self.num_epoch = training_params.get("num_epoch")
+ self.num_epoch = training_params.get("numb_epoch")
168
self.num_epoch_dict = training_params.get("num_epoch_dict")
169
self.disp_file = training_params.get("disp_file", "lcurve.out")
170
self.disp_freq = training_params.get("disp_freq", 1000)
0 commit comments