Merge branch 'main' of github.com:EPCCed/SiMLInt

EPCCed · May 23, 2024 · 7f4be45 · 7f4be45
2 parents c0c66c4 + ff790a3
commit 7f4be45
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
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
diff --git a/files/HW-error-correction/hw.cxx b/files/HW-error-correction/hw.cxx
@@ -131,8 +131,8 @@ class HW : public PhysicsModel {
       client->run_model("hw_model_n", {inKeyN}, {outKeyN});
 
       // unpack output tensor
-      std::vector<double> correctionVort(n_values, 0);
-      std::vector<double> correctionN(n_values, 0);
+      std::vector<float> correctionVort(n_values, 0);
+      std::vector<float> correctionN(n_values, 0);
       client->unpack_tensor(outKeyVort, correctionVort.data(), {n_values},
                             SRTensorTypeFloat, SRMemLayoutContiguous);
       client->unpack_tensor(outKeyN, correctionN.data(), {n_values},