We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 96334fb commit beca0beCopy full SHA for beca0be
1 file changed
deepmd/pt/train/training.py
@@ -1032,7 +1032,7 @@ def log_loss_valid(_task_key="Default"):
1032
with open("checkpoint", "w") as f:
1033
f.write(str(self.latest_model))
1034
1035
- if self.timing_in_training:
+ if self.timing_in_training and self.timed_steps:
1036
msg = f"average training time: {self.total_train_time / self.timed_steps:.4f} s/batch"
1037
excluded_steps = self.num_steps - self.start_step - self.timed_steps
1038
if excluded_steps > 0:
0 commit comments