From 2f786a3551635a864a640c5e3ff0c0e538f72b39 Mon Sep 17 00:00:00 2001 From: Shaochen Shi Date: Fri, 24 Sep 2021 10:44:19 +0800 Subject: [PATCH] Fix freezing error on checkpoint from parallel training. --- deepmd/entrypoints/freeze.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/deepmd/entrypoints/freeze.py b/deepmd/entrypoints/freeze.py index 511f58598d..afbb7659d4 100755 --- a/deepmd/entrypoints/freeze.py +++ b/deepmd/entrypoints/freeze.py @@ -159,6 +159,11 @@ def freeze( clear_devices = True # We import the meta graph and retrieve a Saver + try: + # In case paralle training + import horovod.tensorflow as _ + except ImportError: + pass saver = tf.train.import_meta_graph( f"{input_checkpoint}.meta", clear_devices=clear_devices )