Esempio n. 1
0
 def test_clone(self):
     env = GymEnv(DummyBoxEnv(obs_dim=(4, ), action_dim=(4, )))
     policy = GaussianGRUPolicy(env_spec=env.spec)
     policy_clone = policy.clone('GaussianGRUPolicyClone')
     assert policy_clone.env_spec == policy.env_spec
     for cloned_param, param in zip(policy_clone.parameters.values(),
                                    policy.parameters.values()):
         assert np.array_equal(cloned_param, param)
 def test_clone(self):
     env = GarageEnv(DummyBoxEnv(obs_dim=(4, ), action_dim=(4, )))
     policy = GaussianGRUPolicy(env_spec=env.spec)
     policy_clone = policy.clone('GaussianGRUPolicyClone')
     assert policy_clone.env_spec == policy.env_spec