TensorRT运行TensorFlow pb模型 float16
目录
TensorFlow checkpoint to PB file
TensorFlow checkpoint to PB file
(1)freeze check point to pb file
以bert large为例
freeze_model.py
import tensorflow as tf
# input_checkpoint = './squad_large_output/model.ckpt-43799'
input_checkpoint = './github..
原创
2021-08-12 09:35:27 ·
786 阅读 ·
0 评论