Skip to content

Commit fb4376c

Browse files
authored
Update finetune.rst
1 parent ee63939 commit fb4376c

File tree

1 file changed

+13
-1
lines changed

1 file changed

+13
-1
lines changed

docs/user_guide/finetune.rst

+13-1
Original file line numberDiff line numberDiff line change
@@ -83,4 +83,16 @@ You can replace the data path with your own data path.
8383

8484
.. code-block:: bash
8585
86-
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
86+
torchrun --nproc_per_node=1 src/mattersim/training/finetune_mattersim.py \
87+
--load_model_path mattersim-v1.0.0-1m \
88+
--train_data_path xyz_files/train.xyz \
89+
--valid_data_path xyz_files/valid.xyz \
90+
--batch_size 16 \
91+
--lr 2e-4 \
92+
--step_size 20 \
93+
--epochs 200 \
94+
--save_path ./finetune_result \
95+
--save_checkpoint \
96+
--ckpt_interval 20 \
97+
--include_stresses \
98+
--include_forces

0 commit comments

Comments
 (0)