models.py 文件源码

def get_collection(self, global_step):
    W1 = tf.get_collection("W1")[0]
    W2 = tf.get_collection("W2")[0]
    observations = tf.get_collection("observations")[0]
    probability = tf.get_collection("probability")[0]
    advantages = tf.get_collection("advantages")[0]
    W1Grad = tf.get_collection("W1Grad")[0]
    updateGrads = tf.get_collection("updateGrads")[0]
    W2Grad = tf.get_collection("W2Grad")[0]
    newGrads1 = tf.get_collection("newGrads1")[0]
    newGrads2 = tf.get_collection("newGrads2")[0]
    newGrads = [newGrads1, newGrads2]

    self.global_step = global_step
    self.W1 = W1
    self.W2 = W2
    self.observations = observations
    self.probability = probability
    self.advantages = advantages
    self.W1Grad = W1Grad
    self.updateGrads = updateGrads
    self.W2Grad = W2Grad
    self.newGrads = newGrads

  # Before training, any initialization code