deepmd.tf.utils.multi_init

Module Contents

Functions

replace_model_params_with_frz_multi_model(jdata, ...)	Replace the model params in input script according to pretrained frozen multi-task united model.
_change_sub_config(jdata, src_jdata, sub_key)

Attributes

log

deepmd.tf.utils.multi_init.log

deepmd.tf.utils.multi_init.replace_model_params_with_frz_multi_model(jdata: Dict[str, Any], pretrained_model: str)

Replace the model params in input script according to pretrained frozen multi-task united model.

Parameters:

jdataDict[str, Any]

input script

pretrained_modelstr

filename of the pretrained frozen multi-task united model

deepmd.tf.utils.multi_init._change_sub_config(jdata: Dict[str, Any], src_jdata: Dict[str, Any], sub_key: str)