update readme

geoelements · Aug 26, 2024 · 3222dd9 · 3222dd9
1 parent 06dfb25
commit 3222dd9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -20,7 +20,7 @@ MeshNet is a scalable surrogate simulator for any mesh-based models like Finite
 # For particulate domain,
 python3 -m gns.train mode="train" --config-path ./ --config-name config.yaml
 # For mesh-based domain,
-python3 -m meshnet.train --data_path="<input-training-data-path>" --model_path="<path-to-load-save-model-file>" --ntraining_steps=100
+python3 -m meshnet.train mode="train" --config-path ./ --config-name config_mesh.yaml
 ```
 
 > Resume training
@@ -31,15 +31,15 @@ To resume training specify `model_file` and `train_state_file`:
 # For particulate domain,
 python3 -m gns.train mode="train" training.resume=True
 # For mesh-based domain,
-python3 -m meshnet.train --data_path="<input-training-data-path>" --model_path="<path-to-load-save-model-file>" --model_file="model.pt" --train_state_file="train_state.pt" --ntraining_steps=100
+python3 -m meshnet.train mode="train" training.resume=True
 ```
 
 > Rollout prediction
 ```shell
 # For particulate domain,
 python3 -m gns.train mode="rollout"
 # For mesh-based domain,
-python3 -m meshnet.train --mode="rollout" --data_path="<input-data-path>" --model_path="<path-to-load-save-model-file>" --output_path="<path-to-save-output>" --model_file="model.pt" --train_state_file="train_state.pt"
+python3 -m meshnet.train mode="rollout"
 ```
 
 > Render