def async_initialize(self, agent, sampler_n_itr, batch_spec, mid_batch_reset, examples, world_size=1): self.agent = agent self.n_itr = sampler_n_itr self.batch_spec = batch_spec self.mid_batch_reset = mid_batch_reset self.replay_buffer = initialize_replay_buffer( self, examples, batch_spec, async_=True)
def initialize(self, agent, n_itr, batch_spec, mid_batch_reset, examples, world_size=1, rank=0): self.agent = agent self.n_itr = n_itr self.batch_spec = batch_spec self.mid_batch_reset = mid_batch_reset self.replay_buffer = initialize_replay_buffer(self, examples, batch_spec) self.optim_initialize(rank)