Pytorch迁移到C++

1. 啰嗦的话 首先说说为什么要用pytorch,现在大家都经常用tensorflow、keras、pytorch来编写神经网络,tensorflow和keras属于一路子,现在keras归tensorflow管了,tensorflow的示例程序都是给的keras。如果想在C++端调用写好的模型,则需要保存成pb格式,keras的h5格式是可以转换成pb格式的。基于tensorflow和keras
相关文章
相关标签/搜索