예제 #1
0
 def __init__(self,
              *args,
              observation_key=None,
              desired_goal_key=None,
              **kwargs):
     HER.__init__(
         self,
         observation_key=observation_key,
         desired_goal_key=desired_goal_key,
     )
     SoftActorCritic.__init__(self, *args, **kwargs)
     assert isinstance(self.replay_buffer, ObsDictRelabelingBuffer)
예제 #2
0
파일: her.py 프로젝트: jcoreyes/rlkit
 def __init__(
         self,
         *args,
         her_kwargs,
         sac_kwargs,
         **kwargs
 ):
     HER.__init__(self, **her_kwargs)
     SoftActorCritic.__init__(self, *args, **kwargs, **sac_kwargs)
     assert isinstance(
         self.replay_buffer, RelabelingReplayBuffer
     ) or isinstance(
         self.replay_buffer, ObsDictRelabelingBuffer
     )