1. 首页
  2. 数据库
  3. 其它
  4. 强化学习 —baseline项目之 TensorFlow的训练参数的存储和加载

强化学习 —baseline项目之 TensorFlow的训练参数的存储和加载

上传者: 2021-02-01 14:32:11上传 PDF文件 38.47KB 热度 12次
该项目中把tf的数据存储和读取抽取出两个函数,方便开发,思想和代码值得借迁 一.存储 def save_variables(save_path, variables=None, sess=None): import joblib sess = sess or get_session() variables = variables or tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES) ps = sess.run(variables) save_dict = {v.name: value for v, val
用户评论