a problem when use dp freeze

[lizhiqiang100]

Hello everyone,
After I finish the training process, I ran into a problem when freeze the model as follows: (my version is 2.0.1), Thanks!!

WARNING:tensorflow:From /data/home/lzq21/deepmd-kit/lib/python3.9/site-packages/tensorflow/python/compat/v2_compat.py:96: disable_resource_variables (from tensorflow.python.ops.variable_scope) is deprecated and will be removed in a future version.
Instructions for updating:
non-resource variables are not supported in the long term
WARNING:root:Environment variable KMP_BLOCKTIME is empty. Use the default value 0
WARNING:root:Environment variable KMP_AFFINITY is empty. Use the default value granularity=fine,verbose,compact,1,0
Traceback (most recent call last):
File "/data/home/lzq21/deepmd-kit/bin/dp", line 10, in
sys.exit(main())
File "/data/home/lzq21/deepmd-kit/lib/python3.9/site-packages/deepmd/entrypoints/main.py", line 439, in main
freeze(**dict_args)
File "/data/home/lzq21/deepmd-kit/lib/python3.9/site-packages/deepmd/entrypoints/freeze.py", line 162, in freeze
saver = tf.train.import_meta_graph(
File "/data/home/lzq21/deepmd-kit/lib/python3.9/site-packages/tensorflow/python/training/saver.py", line 1465, in import_meta_graph
return _import_meta_graph_with_return_elements(meta_graph_or_file,
File "/data/home/lzq21/deepmd-kit/lib/python3.9/site-packages/tensorflow/python/training/saver.py", line 1486, in _import_meta_graph_with_return_elements
meta_graph.import_scoped_meta_graph_with_return_elements(
File "/data/home/lzq21/deepmd-kit/lib/python3.9/site-packages/tensorflow/python/framework/meta_graph.py", line 801, in import_scoped_meta_graph_with_return_elements
imported_return_elements = importer.import_graph_def(
File "/data/home/lzq21/deepmd-kit/lib/python3.9/site-packages/tensorflow/python/util/deprecation.py", line 535, in new_func
return func(*args, **kwargs)
File "/data/home/lzq21/deepmd-kit/lib/python3.9/site-packages/tensorflow/python/framework/importer.py", line 400, in import_graph_def
return _import_graph_def_internal(
File "/data/home/lzq21/deepmd-kit/lib/python3.9/site-packages/tensorflow/python/framework/importer.py", line 496, in _import_graph_def_internal
results = c_api.TF_GraphImportGraphDefWithResults(
tensorflow.python.framework.errors_impl.NotFoundError: Op type not registered 'HorovodAllreduce' in binary running on mgt01. Make sure the Op and Kernel are registered in the binary running in this process. Note that if you are loading a saved graph which used ops from tf.contrib, accessing (e.g.) tf.contrib.resampler should be done before importing the graph, as contrib ops are lazily registered when the module is first accessed.

[njzjz]

See #1193.