1453d07070a71f0f0d41a6d9424aa46fc27a5cee,research/learning_to_remember_rare_events/model.py,Model,episode_predict,#Model#Any#Any#Any#Any#,273

Before Change

      y_preds.append(y_pred)

    sess.run([self.mem_reset_op],
             feed_dict={self.mem_keys_reset: cur_memory[0],
                        self.mem_vals_reset: cur_memory[1],
                        self.mem_age_reset: cur_memory[2]})

    return y_preds

After Change

    // Storing current memory state to restore it after prediction
    mem_keys, mem_vals, mem_age, _ = self.memory.get()
    cur_memory = (
        tf.identity(mem_keys),
        tf.identity(mem_vals),
        tf.identity(mem_age),
        None,
    )

    if clear_memory:
      self.clear_memory(sess)

Instances