Fix issue this the path to load the model + new model to take into ac…

…count the new GEMC geometry.
JeffersonLab · Feb 12, 2025 · 992441e · 992441e
1 parent c7c9910
commit 992441e
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/etc/nnet/ALERT/model_AHDC/model_AHDC.pt b/etc/nnet/ALERT/model_AHDC/model_AHDC.pt
diff --git a/reconstruction/alert/src/main/java/org/jlab/rec/ahdc/AI/Model.java b/reconstruction/alert/src/main/java/org/jlab/rec/ahdc/AI/Model.java
@@ -37,7 +37,7 @@ public NDList processInput(TranslatorContext translatorContext, float[] floats)
 
         String path = CLASResources.getResourcePath("etc/nnet/ALERT/model_AHDC/");
         Criteria<float[], Float> my_model = Criteria.builder().setTypes(float[].class, Float.class)
-                .optModelPath(Paths.get("etc/nnet/ALERT/model_AHDC/"))
+                .optModelPath(Paths.get(path))
                 .optEngine("PyTorch")
                 .optTranslator(my_translator)
                 .optProgress(new ProgressBar())