tensorflow 保存模型和取出中间权重例子

下面代码的功能是先训练一个简单的模型，然后保存模型，同时保存到一个pb文件当中，后续可以从pd文件里读取权重值。 import tensorflow as tf import numpy as np import os import h5py import pickle from tensorflow.python.framework import graph_util from tensorflow.python.platform import gfile #设置使用指定GPU os.environ['CUDA_VISIBLE_DEVICES'] = '1' #下面这段代码是在训练好之后将