摘要: tensorflow在保存权重模型时多使用tf.train.Saver().save 函数进行权重保存,保存的ckpt文件无法直接打开,不利于将模型权重导入到其他框架使用(如Caffe、Keras等)。 好在tensorflow提供了相关函数 tf.train.NewCheckpointReader 阅读全文
posted @ 2018-07-03 15:06 张中星 阅读(6175) 评论(0) 推荐(0) 编辑