agent.py 文件源码

python
阅读 34 收藏 0 点赞 0 评论 0

项目:async-rl-tensorflow 作者: datavizweb 项目源码 文件源码
def __init__(self, config, environment, optimizer, lr_op):
    super(Agent, self).__init__(config)
    self.weight_dir = 'weights'

    self.env = environment
    self.history = History(self.config)

    self.lr_op = lr_op
    self.optimizer = optimizer

    self.step_op = tf.Variable(0, trainable=False, name='step')
    self.step_inc_op = self.step_op.assign_add(1, use_locking=True)
    self.build_dqn()

    self.saver = tf.train.Saver(self.w.values() + [self.step_op], max_to_keep=30)
    self.summary_op = tf.merge_all_summaries()
    self.init_op = tf.initialize_all_variables()
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号