def __getstate__(self):
d = Serializable.__getstate__(self)
# Add these explicitly in case they were modified
d["_obs_mean"] = self._obs_mean
d["_obs_std"] = self._obs_std
d["_reward_scale"] = self._reward_scale
return d
def __getstate__(self):
d = Serializable.__getstate__(self)
d["_diayn_disc"] = self.disc.state_dict()
d["_diayn_optim"] = self.disc_optim.state_dict()
d["_n_skills"] = self.n_skills
d["_sample_z"] = self.sample_z
return d
def __getstate__(self):
return Serializable.__getstate__(self)
def __getstate__(self):
d = Serializable.__getstate__(self)
d["params"] = self.get_param_values()
return d
def __getstate__(self):
d = Serializable.__getstate__(self)
return d