Merge pull request #9 from rll-research/proto_fix

fix proto
rll-research · Dec 21, 2021 · bb98f0c · bb98f0c
2 parents a81aed0 + 089e155
commit bb98f0c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/agent/proto.py b/agent/proto.py
@@ -109,7 +109,7 @@ def compute_intr_reward(self, obs, step):
         # enqueue candidates
         ptr = self.queue_ptr
         self.queue[ptr:ptr + self.num_protos] = z[candidates]
-        queue_ptr = (ptr + self.num_protos) % self.queue.shape[0]
+        self.queue_ptr = (ptr + self.num_protos) % self.queue.shape[0]
 
         # compute distances between the batch and the queue of candidates
         z_to_q = torch.norm(z[:, None, :] - self.queue[None, :, :], dim=2, p=2)