`deepmd.tf.nvnmd.entrypoints.freeze`

Module Contents

`filter_tensorVariableList`(→ dict)	Get the name of variable for NVNMD.
`save_weight`(sess[, file_name])	Save the dictionary of weight to a npy file.

deepmd.tf.nvnmd.entrypoints.freeze.filter_tensorVariableList(tensorVariableList) → dict[source]: Get the name of variable for NVNMD.

train_attr/min_nbor_dist

descrpt_attr/t_avg:0

descrpt_attr/t_std:0

type_embed_net/matrix_{layer l}:0

type_embed_net/bias_{layer l}:0

version 0: | filter_type_{atom i}/matrix_{layer l}_{atomj}:0

filter_type_{atom i}/bias_{layer l}_{atomj}:0

layer_{layer l}_type_{atom i}/matrix:0

layer_{layer l}_type_{atom i}/bias:0

final_layer_type_{atom i}/matrix:0

final_layer_type_{atom i}/bias:0

version 1: | filter_type_all/matrix_{layer l}:0

filter_type_all/bias_{layer l}:0

filter_type_all/matrix_{layer l}_two_side_ebd:0

filter_type_all/bias_{layer l}_two_side_ebd:0

layer_{layer l}/matrix:0

layer_{layer l}/bias:0

final_layer/matrix:0

final_layer/bias:0

deepmd.tf.nvnmd.entrypoints.freeze.save_weight(sess, file_name: str = 'nvnmd/weight.npy')[source]: Save the dictionary of weight to a npy file.