d4255c9c4d04cf7f09881b272535cfdc155957a7,env.py,Env,_reset_buffer,#Env#,32

Before Change

  def _reset_buffer(self):
    for _ in range(self.window):
      self.state_buffer.append(self.dtype(84, 84).zero_())

  def reset(self):
    if self.life_termination:

After Change

  def _reset_buffer(self):
    for _ in range(self.window):
      self.state_buffer.append(torch.zeros(84, 84, device=self.device))

  def reset(self):
    if self.life_termination:

Instances