diff --git a/deepmd/main.py b/deepmd/main.py index ec8fbe1dc8..2efd66184e 100644 --- a/deepmd/main.py +++ b/deepmd/main.py @@ -258,7 +258,8 @@ def main_parser() -> argparse.ArgumentParser: parser_train.add_argument( "--use-pretrain-script", action="store_true", - help="(Supported Backend: PyTorch) Use model params in the script of the pretrained model instead of user input when doing finetuning.", + help="(Supported Backend: PyTorch) Use model params in the script of the pretrained model instead of user input when doing finetuning." + "In TensorFlow, it's default and unchangeable to use model params in the script of the pretrained model.", ) parser_train.add_argument( "-o", diff --git a/doc/train/finetuning.md b/doc/train/finetuning.md index f01e3cc759..1cd88191d2 100644 --- a/doc/train/finetuning.md +++ b/doc/train/finetuning.md @@ -29,8 +29,8 @@ The command above will change the energy bias in the last layer of the fitting n according to the training dataset in input.json. :::{warning} -Note that in TensorFlow, the `type_map` will be overwritten based on that in the pre-trained model. -Please ensure you are familiar with the `type_map` configuration in the pre-trained model before starting the fine-tuning process. +Note that in TensorFlow, model parameters including the `type_map` will be overwritten based on those in the pre-trained model. +Please ensure you are familiar with the configurations in the pre-trained model, especially `type_map`, before starting the fine-tuning process. The elements in the training dataset must be contained in the pre-trained dataset. :::