m0_37855677的博客

原创 Pytorch迁移到C++

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