From 8fa3a30107095ae92fdf16ce9f297e452ac60ebe Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Mon, 2 May 2022 16:08:59 -0400 Subject: [PATCH] provide valid_data the same type_map as train_data Fix #1676. --- deepmd/entrypoints/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepmd/entrypoints/train.py b/deepmd/entrypoints/train.py index 42e21d2ba4..4a67eee9d4 100755 --- a/deepmd/entrypoints/train.py +++ b/deepmd/entrypoints/train.py @@ -154,7 +154,7 @@ def _do_work(jdata: Dict[str, Any], run_opt: RunOptions, is_compress: bool = Fal train_data = get_data(jdata["training"]["training_data"], rcut, ipt_type_map, modifier) train_data.print_summary("training") if jdata["training"].get("validation_data", None) is not None: - valid_data = get_data(jdata["training"]["validation_data"], rcut, ipt_type_map, modifier) + valid_data = get_data(jdata["training"]["validation_data"], rcut, train_data.type_map, modifier) valid_data.print_summary("validation") # get training info