微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

从Java中的PyTorch jit模型中提取ExtraFileMap信息

如何解决从Java中的PyTorch jit模型中提取ExtraFileMap信息

我可以使用PyTorch Java Bindings库在Java中运行以Python原生训练的Pytorch模型。太棒了,但是,我无法提取与模型关联的任何元数据?

保存经过训练的模型时,可以使用ExtraFilesMap对象在Python或C ++中添加额外的信息。在这种情况下,额外信息是模型在其上进行训练的音频数据的类型。这对于在将音频数据传递到模型进行分类之前进行预处理至关重要。在Python中,可以通过...从保存的模型中提取额外的元数据。

extra_files = torch._C.ExtraFilesMap()
model = torch.jit.load(filepath_model,_extra_files=extra_files)
dataOptsstr = extra_files['dataOpts']

在Java中,可以使用...加载经过训练的模型。

Module mod = Module.load(modelPath);

但是,似乎没有任何方法可以提取ExtraFileMap信息。我花了很长时间试图找到一个没有运气的解决方案。这里缺少什么吗?似乎是在不同语言之间共享模型的重要功能

完整的代码示例代码可在BatDL.java here中找到。

如果使用当前的Pytorch Java库确实无法执行此操作,那么有人对在Java中如何完成它有任何建议吗?

在此先感谢您的帮助。

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。