Skip to content

Commit

Permalink
add doc for use-pretrain-script in tf
Browse files Browse the repository at this point in the history
  • Loading branch information
iProzd committed Jun 10, 2024
1 parent 4599213 commit bf20853
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 3 deletions.
3 changes: 2 additions & 1 deletion deepmd/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down
4 changes: 2 additions & 2 deletions doc/train/finetuning.md
Original file line number Diff line number Diff line change
Expand Up @@ -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.
:::

Expand Down

0 comments on commit bf20853

Please sign in to comment.