|
|
|
|
|
|
self.training_buffer[agent_id]['visual_obs%d' % i].append( |
|
|
|
stored_info.visual_observations[i][idx]) |
|
|
|
self.training_buffer[agent_id]['next_visual_obs%d' % i].append( |
|
|
|
next_info.visual_observations[i][idx]) |
|
|
|
next_info.visual_observations[i][next_idx]) |
|
|
|
if self.use_vector_obs: |
|
|
|
self.training_buffer[agent_id]['vector_obs'].append(stored_info.vector_observations[idx]) |
|
|
|
self.training_buffer[agent_id]['next_vector_in'].append( |
|
|
|