From d05e27b3a87b53c29c004f162915421e5b292616 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Tue, 5 Mar 2024 23:36:26 -0500 Subject: [PATCH] pt: fix print_on_training when there is no validation data Signed-off-by: Jinzhe Zeng #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 e5a7632ac4..78f56e4be4 100644 --- a/deepmd/pt/train/training.py +++ b/deepmd/pt/train/training.py @@ -1016,7 +1016,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]) @@ -1026,7 +1026,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 % (