def __setstate__(self, d):
Serializable.__setstate__(self, d)
self._obs_mean = d["_obs_mean"]
self._obs_std = d["_obs_std"]
self._reward_scale = d["_reward_scale"]
def __setstate__(self, state):
return Serializable.__setstate__(self, state)
def __setstate__(self, d):
Serializable.__setstate__(self, d)
self.set_param_values(d["params"])
def __setstate__(self, d):
Serializable.__setstate__(self, d)
def __setstate__(self, d):
Serializable.__setstate__(self, d)
self.disc.load_state_dict(d["_diayn_disc"])
self.disc_optim.load_state_dict(d["_diayn_optim"])
self.n_skills = d["_n_skills"]
self.sample_z = d["_sample_z"]
def __setstate__(self, d):
Serializable.__setstate__(self, d)
self._obs_absmax = d["_obs_absmax"]
self._reward_scale = d["_reward_scale"]