Pytorch遷移到C++

1. 囉嗦的話 首先說說爲什麼要用pytorch,現在大家都經常用tensorflow、keras、pytorch來編寫神經網絡,tensorflow和keras屬於一路子,現在keras歸tensorflow管了,tensorflow的示例程序都是給的keras。如果想在C++端調用寫好的模型,則需要保存成pb格式,keras的h5格式是可以轉換成pb格式的。基於tensorflow和keras
相關文章
相關標籤/搜索