deepmd.tf.nvnmd.utils.config

Module Contents

Classes

NvnmdConfig

Configuration for NVNMD

Attributes

log
nvnmd_cfg

deepmd.tf.nvnmd.utils.config.log

class deepmd.tf.nvnmd.utils.config.NvnmdConfig(jdata: dict)

Configuration for NVNMD record the message of model such as size, using nvnmd or not.

Parameters:

jdata

a dictionary of input script

References

DOI: 10.1038/s41524-022-00773-z

init_from_jdata(jdata: dict = {})

Initialize this class with jdata loaded from input script.

init_value()

Initialize member with dict.

update_config()

Update config from dict.

init_train_mode(mod='cnn')

Configure for taining cnn or qnn.

init_from_config(jdata)

Initialize member element one by one.

init_config_by_version(version, max_nnei)

Initialize version-dependent parameters.

init_net_size()

Initialize net_size.

init_from_deepmd_input(jdata)

Initialize members with input script of deepmd.

init_dscp(jdata: dict, jdata_parent: dict = {}) → dict

Initialize members about descriptor.

init_fitn(jdata: dict, jdata_parent: dict = {}) → dict

Initialize members about fitting network.

init_dpin(jdata: dict, jdata_parent: dict = {}) → dict

Initialize members about other deepmd input.

init_size(jdata: dict, jdata_parent: dict = {}) → dict

Initialize members about ram capacity.

init_ctrl(jdata: dict, jdata_parent: dict = {}) → dict

Initialize members about control signal.

init_nbit(jdata: dict, jdata_parent: dict = {}) → dict

Initialize members about quantification precision.

save(file_name=None)

Save all configuration to file.

set_ntype(ntype)

Set the number of type.

get_s_range(davg, dstd)

Get the range of switch function.

get_dscp_jdata()

Generate model/descriptor in input script.

get_fitn_jdata()

Generate model/fitting_net in input script.

get_model_jdata()

Generate model in input script.

get_nvnmd_jdata()

Generate nvnmd in input script.

get_learning_rate_jdata()

Generate learning_rate in input script.

get_loss_jdata()

Generate loss in input script.

get_training_jdata()

Generate training in input script.

get_deepmd_jdata()

Generate input script with member element one by one.

get_dp_init_weights()

Build the weight dict for initialization of net.

disp_message()

Display the log of NVNMD.

deepmd.tf.nvnmd.utils.config.nvnmd_cfg