[dp-compress] min_nbor_dist error

[AnuragKr]

Hi,

I hope you are doing well. I am getting errors when I am doing model compression.

DeepMD-kit Version -- v2.1.3

Input File

{
    "_comment": " model parameters",
    "model": {
	"type_map":	["O", "H"],
	"descriptor" :{
	    "type":		"se_e2_a",
	    "sel":		[46, 92],
	    "rcut_smth":	0.50,
	    "rcut":		6.00,
	    "neuron":		[32,64,128],
	    "resnet_dt":	false,
	    "axis_neuron":	16,
	    "seed":		1,
	    "_comment":		" that's all"
	},
	"fitting_net" : {
	    "neuron":		[240, 240,240],
	    "resnet_dt":	true,
	    "seed":		1,
	    "_comment":		" that's all"
	},
	"_comment":	" that's all"
    },

    "learning_rate" :{
	"type":		"exp",
	"decay_steps":	5000,
	"start_lr":	0.001,	
	"stop_lr":	3.51e-8,
	"_comment":	"that's all"
    },

    "loss" :{
	"type":		"ener",
	"start_pref_e":	0.02,
	"limit_pref_e":	8,
	"start_pref_f":	1000,
	"limit_pref_f":	1,
	"start_pref_v":	0,
	"limit_pref_v":	0,
	"_comment":	" that's all"
    },

    "training" : {
	"training_data": {
	    "systems":		 "../../data/water/lw_pimd/train",
	    "batch_size":	"auto",
	    "_comment":		"that's all"
	},
	"validation_data":{
	    "systems":		"../../data/water/lw_pimd/val",
	    "batch_size":	"auto",
	    "_comment":		"that's all"
	},
	"numb_steps":	1000000,
	"seed":		10,
	"disp_file":	"lcurve.out",
	"disp_freq":	100,
	"save_freq":	1000,
	"_comment":	"that's all"
    },    

    "_comment":		"that's all"
}

Command --

dp compress -i graph.pb -o graph-compress.pb

Error--

WARNING:tensorflow:From /home/anurag1/miniconda3/envs/deepmd/lib/python3.10/site-packages/tensorflow/python/compat/v2_compat.py:107: 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
/home/anurag1/miniconda3/envs/deepmd/lib/python3.10/importlib/init.py:169: UserWarning: The NumPy module was reloaded (imported a second time). This can in some cases result in small but subtle issues and is discouraged.
_bootstrap._exec(spec, module)
Traceback (most recent call last):
File "/home/anurag1/miniconda3/envs/deepmd/lib/python3.10/site-packages/deepmd/utils/graph.py", line 56, in get_tensor_by_name_from_graph
tensor = graph.get_tensor_by_name(tensor_name + ':0')
File "/home/anurag1/miniconda3/envs/deepmd/lib/python3.10/site-packages/tensorflow/python/framework/ops.py", line 4128, in get_tensor_by_name
return self.as_graph_element(name, allow_tensor=True, allow_operation=False)
File "/home/anurag1/miniconda3/envs/deepmd/lib/python3.10/site-packages/tensorflow/python/framework/ops.py", line 3952, in as_graph_element
return self._as_graph_element_locked(obj, allow_tensor, allow_operation)
File "/home/anurag1/miniconda3/envs/deepmd/lib/python3.10/site-packages/tensorflow/python/framework/ops.py", line 3992, in _as_graph_element_locked
raise KeyError("The name %s refers to a Tensor which does not "
KeyError: "The name 'train_attr/min_nbor_dist:0' refers to a Tensor which does not exist. The operation, 'train_attr/min_nbor_dist', does not exist in the graph."

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
File "/home/anurag1/miniconda3/envs/deepmd/lib/python3.10/site-packages/deepmd/entrypoints/compress.py", line 71, in compress
t_min_nbor_dist = get_tensor_by_name(input, 'train_attr/min_nbor_dist')
File "/home/anurag1/miniconda3/envs/deepmd/lib/python3.10/site-packages/deepmd/utils/graph.py", line 87, in get_tensor_by_name
return get_tensor_by_name_from_graph(graph, tensor_name)
File "/home/anurag1/miniconda3/envs/deepmd/lib/python3.10/site-packages/deepmd/utils/graph.py", line 58, in get_tensor_by_name_from_graph
raise GraphWithoutTensorError() from e
deepmd.utils.errors.GraphWithoutTensorError

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
File "/home/anurag1/miniconda3/envs/deepmd/bin/dp", line 10, in
sys.exit(main())
File "/home/anurag1/miniconda3/envs/deepmd/lib/python3.10/site-packages/deepmd/entrypoints/main.py", line 526, in main
compress(**dict_args)
File "/home/anurag1/miniconda3/envs/deepmd/lib/python3.10/site-packages/deepmd/entrypoints/compress.py", line 75, in compress
raise RuntimeError(
RuntimeError: The input frozen model: graph.pb has no training script or min_nbor_dist information, which is not supported by the model compression interface. Please consider using the --training-script command within the model compression interface to provide the training script of the input frozen model. Note that the input training script must contain the correct path to the training data.

[njzjz]

The error message itself has already explained the reason and what to do.

[AnuragKr]

Thanks for pointing it out, I completely missed it.