search

bioimage-io / spec-bioimage-io

Specification for the bioimage.io model description file.
https://bioimage-io.github.io/spec-bioimage-io/
MIT License
18 stars 16 forks source link

TorchScript model deepImageJ #128

Closed esgomezm closed 3 years ago

esgomezm commented 3 years ago

Hi, here is the error given by Fiji:

ai.djl.engine.EngineException: [enforce fail at inline_container.cc:222] . file not found: archive/constants.pkl
at ai.djl.pytorch.jni.PyTorchLibrary.moduleLoad(Native Method)
at ai.djl.pytorch.jni.JniUtils.loadModule(JniUtils.java:1247)
at ai.djl.pytorch.engine.PtModel.load(PtModel.java:78)
at ai.djl.repository.zoo.BaseModelLoader.loadModel(BaseModelLoader.java:142)
at ai.djl.repository.zoo.ModelZoo.loadModel(ModelZoo.java:169)
at deepimagej.stamp.LoadPytorchStamp.run(LoadPytorchStamp.java:231)
at java.lang.Thread.run(Thread.java:748)

@constantinpape, could you please try to export it using PyTorch 1.6?

constantinpape commented 3 years ago

We have working torchscript models now, so I'm gonna close this one.