Cyril的博客

原创 将tensorflow保存的预训练模型读取为pytorch模型参数

用tensorflow的saver保存模型后，会有如下几个文件 我们可以通过graph.pbtxt看到tensorflow的计算图，方法如下： import tensorflow as tf import tensorflow.contrib.image from tensorflow.python.platform import gfile graph = tf.get_default_gra...