deepmd.tf.nvnmd.entrypoints.freeze

Functions

filter_tensorVariableList(→ dict)	Get the name of variable for NVNMD.
save_weight(→ None)	Save the dictionary of weight to a npy file.

Module Contents

deepmd.tf.nvnmd.entrypoints.freeze.filter_tensorVariableList(tensorVariableList) → dict

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') → None

Save the dictionary of weight to a npy file.