def test_clone(self): env = GarageEnv(DummyBoxEnv(obs_dim=(2, ), action_dim=(2, ))) embedding_spec = InOutSpec(input_space=env.spec.observation_space, output_space=env.spec.action_space) embedding = GaussianMLPEncoder(embedding_spec) clone_embedding = embedding.clone(name='cloned') assert clone_embedding.input_dim == embedding.input_dim assert clone_embedding.output_dim == embedding.output_dim
def test_clone(self): env = GymEnv(DummyBoxEnv(obs_dim=(2, ), action_dim=(2, ))) embedding_spec = InOutSpec(input_space=env.spec.observation_space, output_space=env.spec.action_space) embedding = GaussianMLPEncoder(embedding_spec) clone_embedding = embedding.clone(name='cloned') assert clone_embedding.input_dim == embedding.input_dim for cloned_param, param in zip( clone_embedding.model.parameters.values(), embedding.model.parameters.values()): assert np.array_equal(cloned_param, param) assert clone_embedding.output_dim == embedding.output_dim