From 61398088254214bea453e42a17202ffdfb5d1c04 Mon Sep 17 00:00:00 2001 From: Amrey Krause Date: Thu, 23 May 2024 11:22:03 +0100 Subject: [PATCH 1/2] Train the model for phi as well --- files/5-training/submit-training.sh | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/files/5-training/submit-training.sh b/files/5-training/submit-training.sh index c1a0216..1163d59 100644 --- 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 \ No newline at end of file + --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 From ff790a3a0fac4f613c213401c9a775e2b756b596 Mon Sep 17 00:00:00 2001 From: Amrey Krause Date: Thu, 23 May 2024 11:32:08 +0100 Subject: [PATCH 2/2] Unpack error correction tensor into float array not double --- files/HW-error-correction/hw.cxx | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/files/HW-error-correction/hw.cxx b/files/HW-error-correction/hw.cxx index ce14bc1..004a8aa 100644 --- 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 correctionVort(n_values, 0); - std::vector correctionN(n_values, 0); + std::vector correctionVort(n_values, 0); + std::vector correctionN(n_values, 0); client->unpack_tensor(outKeyVort, correctionVort.data(), {n_values}, SRTensorTypeFloat, SRMemLayoutContiguous); client->unpack_tensor(outKeyN, correctionN.data(), {n_values},