def test_clone(self):
     env = GarageEnv(DummyDiscreteEnv(obs_dim=(10, ), action_dim=4))
     policy = CategoricalGRUPolicy(env_spec=env.spec)
     policy_clone = policy.clone('CategoricalGRUPolicyClone')
     assert policy.env_spec == policy_clone.env_spec
     for cloned_param, param in zip(policy_clone.parameters.values(),
                                    policy.parameters.values()):
         assert np.array_equal(cloned_param, param)
示例#2
0
 def test_clone(self):
     env = GarageEnv(DummyDiscreteEnv(obs_dim=(10, ), action_dim=4))
     policy = CategoricalGRUPolicy(env_spec=env.spec)
     policy_clone = policy.clone('CategoricalGRUPolicyClone')
     assert policy.env_spec == policy_clone.env_spec