def __setstate__(self, d): Serializable.__setstate__(self, d) global load_params if load_params: tf.get_default_session().run( tf.variables_initializer(self.get_params())) self.set_param_values(d["params"])
def __setstate__(self, d): """Set Serializable state fo the RLAlgorithm instance.""" Serializable.__setstate__(self, d) self._qf1.set_param_values(d['qf1-params']) self._qf2.set_param_values(d['qf2-params']) self._vf.set_param_values(d['vf-params']) self._policy.set_param_values(d['policy-params']) self._pool.__setstate__(d['pool']) self._env.__setstate__(d['env'])