Train the model for phi as well

EPCCed · May 23, 2024 · 6139808 · 6139808
1 parent 397c6a3
commit 6139808
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/files/5-training/submit-training.sh b/files/5-training/submit-training.sh
@@ -30,4 +30,8 @@ python ${SIMLINT_HOME}/files/5-training/training.py --epochs 100 --batch-size 32
 
 # model the density error
 python ${SIMLINT_HOME}/files/5-training/training.py --epochs 100 --batch-size 32 --learning-rate 0.0001 \
-    --trajectories 10 --data-directory ${WORK}/data/training/ --variables n --task-id dens
+    --trajectories 10 --data-directory ${WORK}/data/training/ --variables n --task-id dens
+
+# model the electrostatic potential error
+python ${SIMLINT_HOME}/files/5-training/training.py --epochs 100 --batch-size 32 --learning-rate 0.0001 \
+    --trajectories 10 --data-directory ${WORK}/data/training/ --variables phi --task-id phi