pytorch / java-demo

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Does this binding for java a full version?

rayfalling opened this issue · comments

Can I train model by using java only?
Currently my team is using swig to warpper libtorch c++ frontend to java.

No. The current bindings are for inference only.

@rayfalling That's something we can now do with the JavaCPP Presets for PyTorch though:
https://github.com/bytedeco/javacpp-presets/blob/master/pytorch/samples/SimpleMNIST.java
Please refer to pytorch/pytorch#58973 for a discussion about that.