From fb4376c8ada37795a7267e83cd5205fef1b291f4 Mon Sep 17 00:00:00 2001 From: Han Yang Date: Wed, 15 Jan 2025 13:26:31 +0800 Subject: [PATCH] Update finetune.rst --- docs/user_guide/finetune.rst | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/docs/user_guide/finetune.rst b/docs/user_guide/finetune.rst index 9ef6ad0..e2ccef8 100644 --- a/docs/user_guide/finetune.rst +++ b/docs/user_guide/finetune.rst @@ -83,4 +83,16 @@ You can replace the data path with your own data path. .. code-block:: bash - torchrun --nproc_per_node=1 src/mattersim/training/finetune_mattersim.py --load_model_path mattersim-v1.0.0-1m --train_data_path xyz_files/train.xyz --valid_data_path xyz_files/valid.xyz --batch_size 16 --lr 2e-4 --step_size 20 --epochs 200 --save_path ./finetune_result --save_checkpoint --ckpt_interval 20 --include_stresses --include_forces + torchrun --nproc_per_node=1 src/mattersim/training/finetune_mattersim.py \ + --load_model_path mattersim-v1.0.0-1m \ + --train_data_path xyz_files/train.xyz \ + --valid_data_path xyz_files/valid.xyz \ + --batch_size 16 \ + --lr 2e-4 \ + --step_size 20 \ + --epochs 200 \ + --save_path ./finetune_result \ + --save_checkpoint \ + --ckpt_interval 20 \ + --include_stresses \ + --include_forces