Python GaussianMLPEncoder.clone Examples

Example #1

File: test_gaussian_mlp_encoder.py Project: j-donahue/garage

 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

Example #2

File: test_gaussian_mlp_encoder.py Project: ziyiwu9494/garage

 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