File tree Expand file tree Collapse file tree 1 file changed +13
-1
lines changed Expand file tree Collapse file tree 1 file changed +13
-1
lines changed Original file line number Diff line number Diff line change @@ -83,4 +83,16 @@ You can replace the data path with your own data path.
83
83
84
84
.. code-block :: bash
85
85
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
You can’t perform that action at this time.
0 commit comments