deepmd.tf.nvnmd.entrypoints.freeze
Module Contents
Functions
| Get the name of variable for NVNMD. |
| 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