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
- version = 0[source]
- enable = False[source]
- map[source]
- config[source]
- save_path = 'nvnmd/config.npy'[source]
- weight[source]
- init_from_jdata(jdata: dict = {}) → None[source]
Initialize this class with jdata loaded from input script.
- init_value() → None[source]
Initialize member with dict.
- update_config() → None[source]
Update config from dict.
- init_train_mode(mod='cnn') → None[source]
Configure for taining cnn or qnn.
- init_from_config(jdata) → None[source]
Initialize member element one by one.
- init_config_by_version(version, max_nnei) → None[source]
Initialize version-dependent parameters.
- init_net_size() → None[source]
Initialize net_size.
- init_from_deepmd_input(jdata) → None[source]
Initialize members with input script of deepmd.
- init_dscp(jdata: dict, jdata_parent: dict = {}) → dict[source]
Initialize members about descriptor.
- init_fitn(jdata: dict, jdata_parent: dict = {}) → dict[source]
Initialize members about fitting network.
- init_dpin(jdata: dict, jdata_parent: dict = {}) → dict[source]
Initialize members about other deepmd input.
- init_size(jdata: dict, jdata_parent: dict = {}) → dict[source]
Initialize members about ram capacity.
- init_ctrl(jdata: dict, jdata_parent: dict = {}) → dict[source]
Initialize members about control signal.
- init_nbit(jdata: dict, jdata_parent: dict = {}) → dict[source]
Initialize members about quantification precision.
- save(file_name=None) → None[source]
Save all configuration to file.
- set_ntype(ntype) → None[source]
Set the number of type.
- get_s_range(davg, dstd) → None[source]
Get the range of switch function.
- get_dscp_jdata()[source]
Generate model/descriptor in input script.
- get_fitn_jdata()[source]
Generate model/fitting_net in input script.
- get_model_jdata()[source]
Generate model in input script.
- get_nvnmd_jdata()[source]
Generate nvnmd in input script.
- get_learning_rate_jdata()[source]
Generate learning_rate in input script.
- get_loss_jdata()[source]
Generate loss in input script.
- get_training_jdata()[source]
Generate training in input script.
- get_deepmd_jdata()[source]
Generate input script with member element one by one.
- get_dp_init_weights()[source]
Build the weight dict for initialization of net.
- disp_message() → None[source]
Display the log of NVNMD.
