From fc1cfd27d5615ad0d39c945bbd8e8a2473b1d43d Mon Sep 17 00:00:00 2001 From: carlosuc3m <100329787@alumnos.uc3m.es> Date: Sat, 23 Nov 2024 16:45:58 +0100 Subject: [PATCH] try more prints --- .../bioimage/modelrunner/pytorch/javacpp/JavaWorker.java | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/JavaWorker.java b/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/JavaWorker.java index b46f7c7..b696fb0 100644 --- a/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/JavaWorker.java +++ b/src/main/java/io/bioimage/modelrunner/pytorch/javacpp/JavaWorker.java @@ -35,6 +35,7 @@ public static void main(String[] args) { try { pi = new PytorchJavaCPPInterface(false); } catch (IOException | URISyntaxException e) { + e.printStackTrace(); return; } @@ -87,13 +88,16 @@ private void executeScript(String script, Map inputs) { this.reportLaunch(); try { if (script.equals("loadModel")) { - update("STATY IN WORKER LOAD", null, null); + update("STATY IN WORKER LOAD LOAD", null, null); pi.loadModel((String) inputs.get("modelFolder"), (String) inputs.get("modelSource")); } else if (script.equals("inference")) { - update("STATY IN WORKER", null, null); + update("STATY IN WORKER ------------RUN", null, null); pi.runFromShmas((List) inputs.get("inputs"), (List) inputs.get("outputs")); } else if (script.equals("close")) { pi.closeModel(); + } else { + update("LOL WTF", null, null); + update("LOL WTF -------------- " + script, null, null); } } catch(Exception | Error ex) { this.fail(Types.stackTrace(ex));