deepmd.tf.utils.finetune

Module Contents

Functions

replace_model_params_with_pretrained_model(jdata, ...)

Replace the model params in input script according to pretrained model.

Attributes

log

deepmd.tf.utils.finetune.log

deepmd.tf.utils.finetune.replace_model_params_with_pretrained_model(jdata: Dict[str, Any], pretrained_model: str)

Replace the model params in input script according to pretrained model.

Parameters:

jdataDict[str, Any]

input script

pretrained_modelstr

filename of the pretrained model