From 5ac6729f7e0d79567486e06e97db659f3a964334 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Sun, 22 Sep 2024 01:15:01 -0700 Subject: [PATCH] fix(pt): fix `compute_output_stats_global` when `atomic_output` is `None` --- deepmd/pt/utils/stat.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepmd/pt/utils/stat.py b/deepmd/pt/utils/stat.py index 6de70eb17..58e02f436 100644 --- a/deepmd/pt/utils/stat.py +++ b/deepmd/pt/utils/stat.py @@ -478,7 +478,7 @@ def compute_output_stats_global( std_atom_e = {} for kk in keys: if kk in stats_input: - if atomic_output.get_data()[kk].intensive: + if atomic_output is not None and atomic_output.get_data()[kk].intensive: task_dim = stats_input[kk].shape[1] assert merged_natoms[kk].shape == (nf[kk], ntypes) stats_input[kk] = (