File tree 1 file changed +4
-3
lines changed
src/main/java/io/bioimage/modelrunner/pytorch
1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -245,8 +245,8 @@ private static String getModelName(String modelSource) throws IOException {
245
245
}
246
246
247
247
@ Override
248
- public <T extends RealType <T > & NativeType <T >, R extends RealType <R > & NativeType <R >> List < RandomAccessibleInterval < R >> inference (
249
- List <RandomAccessibleInterval <T >> inputs ) throws RunModelException {
248
+ public <T extends RealType <T > & NativeType <T >, R extends RealType <R > & NativeType <R >>
249
+ List < RandomAccessibleInterval < R >> inference ( List <RandomAccessibleInterval <T >> inputs ) throws RunModelException {
250
250
if (interprocessing ) {
251
251
return runInterprocessing (inputs );
252
252
}
@@ -523,7 +523,8 @@ private void closeShmas() throws RunModelException {
523
523
try { shm .close (); } catch (IOException e1 ) { e1 .printStackTrace ();}
524
524
});
525
525
shmaOutputList = null ;
526
- closeInterprocess ();
526
+ if (interprocessing )
527
+ closeInterprocess ();
527
528
}
528
529
529
530
You can’t perform that action at this time.
0 commit comments