From 6fc64648a4b8f005bd6d7dcc072caab9d046308b Mon Sep 17 00:00:00 2001 From: yanghan234 Date: Wed, 15 Jan 2025 05:35:55 +0000 Subject: [PATCH] deploy: d19370dba20998f9632a0ed72ad29e0419c6fdb2 --- _sources/user_guide/finetune.rst | 14 +++++++++++++- user_guide/finetune.html | 14 +++++++++++++- 2 files changed, 26 insertions(+), 2 deletions(-) diff --git a/_sources/user_guide/finetune.rst b/_sources/user_guide/finetune.rst index 9ef6ad0..e2ccef8 100644 --- a/_sources/user_guide/finetune.rst +++ b/_sources/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 diff --git a/user_guide/finetune.html b/user_guide/finetune.html index ec1e927..db0e095 100644 --- a/user_guide/finetune.html +++ b/user_guide/finetune.html @@ -381,7 +381,19 @@

Finetune Parameters

Finetune Example#

You can replace the data path with your own data path.

-
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