diff --git a/deepmd_pt/utils/stat.py b/deepmd_pt/utils/stat.py index b183bb55..3b0af52c 100644 --- a/deepmd_pt/utils/stat.py +++ b/deepmd_pt/utils/stat.py @@ -83,7 +83,7 @@ def make_stat_input(datasets, dataloaders, nbatches): return lst -def compute_output_stats(energy, natoms, rcond=1e-3): +def compute_output_stats(energy, natoms, rcond=None): """Update mean and stddev for descriptor elements. Args: