ckpt为savermodel模型并TRT调用的问题
因为任务需要,需要将TensorFlow训练的得到的ckpt文件转换成savemodel,并用TRT调用,这里记下自己从中遇到的坑。1、ckpt转savemodel,可以参考https://blog.csdn.net/zmlovelx/article/details/100511406这个博客,写的很详细,按照使用没有任何问题。保存好以后到save目录下,会有一个saved_model.pb文件...
原创
2019-10-23 09:51:50 ·
2291 阅读 ·
8 评论