在Java中部署pytorch模型

时间:2020-03-17 11:50:10

标签: java pytorch

我已经训练并保存了一个pytorch模型,现在我想在Windows OS的Java(非android)环境中使用它(因为我使用的是Java中仅提供的某些库),这可能吗?我在pytorch文档中找不到直接的答案,并且当单击Java api文档时,链接已断开。

1 个答案:

答案 0 :(得分:1)

@Gilad,您可以使用Deep Java Library(djl.ai)进行此操作。检出:https://github.com/awslabs/djl/tree/master/pytorch/pytorch-engine