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)
