From fefc0e6cf64e3dcd6d78e9ce8707f4fc8c2a3b17 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Fri, 8 Mar 2024 03:25:01 -0500 Subject: [PATCH] pt: fix print_on_training when there is no validation data (#3423) #3405 changed results from `None` to `{}` but `print_on_training` wasn't revised. --- deepmd/pt/train/training.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/deepmd/pt/train/training.py b/deepmd/pt/train/training.py index 93afc38575..62bc5a4c97 100644 --- a/deepmd/pt/train/training.py +++ b/deepmd/pt/train/training.py @@ -1037,7 +1037,7 @@ def print_on_training(self, fout, step_id, cur_lr, train_results, valid_results) print_str = "" print_str += "%7d" % step_id if not self.multi_task: - if valid_results is not None: + if valid_results: prop_fmt = " %11.2e %11.2e" for k in train_keys: print_str += prop_fmt % (valid_results[k], train_results[k]) @@ -1047,7 +1047,7 @@ def print_on_training(self, fout, step_id, cur_lr, train_results, valid_results) print_str += prop_fmt % (train_results[k]) else: for model_key in self.model_keys: - if valid_results[model_key] is not None: + if valid_results[model_key]: prop_fmt = " %11.2e %11.2e" for k in sorted(valid_results[model_key].keys()): print_str += prop_fmt % (