Skip to content

Commit a307dbd

Browse files
mathieuouillonbaltzell
authored andcommitted
Update model path for TrackMatchingAI
1 parent e7a70b0 commit a307dbd

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

‎reconstruction/alert/src/main/java/org/jlab/rec/alert/TrackMatchingAI/ModelTrackMatching.java‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ public NDList processInput(TranslatorContext translatorContext, float[] floats)
5252
System.setProperty("ai.djl.pytorch.num_threads", "1");
5353
System.setProperty("ai.djl.pytorch.graph_optimizer", "false");
5454

55-
String path = CLASResources.getResourcePath("etc/data/nnet/ALERT/model_TM/");
55+
String path = CLASResources.getResourcePath("etc/data/nnet/rg-l/model_TM/");
5656
Criteria<float[], float[]> my_model = Criteria.builder().setTypes(float[].class, float[].class)
5757
.optModelPath(Paths.get(path))
5858
.optEngine("PyTorch")

0 commit comments

Comments
 (0)