Move AI model from reconstruction/alert to etc/nnet.

Use CLASResources class to get the path of the model.
JeffersonLab · Jan 10, 2025 · 5c6be95 · 5c6be95
1 parent ea80f18
commit 5c6be95
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/.../java/org/jlab/rec/ahdc/AI/model/model.pt → etc/nnet/ALERT/model_AHDC/model_AHDC.pt b/.../java/org/jlab/rec/ahdc/AI/model/model.pt → etc/nnet/ALERT/model_AHDC/model_AHDC.pt
diff --git a/reconstruction/alert/src/main/java/org/jlab/rec/service/AHDCEngine.java b/reconstruction/alert/src/main/java/org/jlab/rec/service/AHDCEngine.java
@@ -38,6 +38,7 @@
 import org.jlab.rec.ahdc.PreCluster.PreCluster;
 import org.jlab.rec.ahdc.PreCluster.PreClusterFinder;
 import org.jlab.rec.ahdc.Track.Track;
+import org.jlab.utils.CLASResources;
 
 import java.io.File;
 import java.io.IOException;
@@ -80,9 +81,10 @@ public NDList processInput(TranslatorContext translatorContext, float[] floats)
 				return new NDList(samples);
 			}
 		};
-
+
+		String path = CLASResources.getResourcePath("etc/nnet/ALERT/model_AHDC/"); 
 		Criteria<float[], Float> my_model = Criteria.builder().setTypes(float[].class, Float.class)
-				.optModelPath(Paths.get(System.getenv("CLAS12DIR") + "/../reconstruction/alert/src/main/java/org/jlab/rec/ahdc/AI/model/"))
+				.optModelPath(Paths.get(path))
 				.optEngine("PyTorch")
 				.optTranslator(my_translator)
 				.optProgress(new ProgressBar())