(self)
| 229 | |
| 230 | |
| 231 | def load(self): |
| 232 | params = [t for t in tf.trainable_variables() if t.name.startswith(self.scope)] |
| 233 | npz = np.load('tf_dqn_weights.npz') |
| 234 | ops = [] |
| 235 | for p, (_, v) in zip(params, npz.iteritems()): |
| 236 | ops.append(p.assign(v)) |
| 237 | self.session.run(ops) |
| 238 | |
| 239 | |
| 240 | def set_session(self, session): |