Update

[ghstack-poisoned]
pytorch · Dec 15, 2024 · 5ffd212 · 5ffd212
1 parent 65bb4f3
commit 5ffd212
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sota-implementations/iql/iql_online.py b/sota-implementations/iql/iql_online.py
@@ -164,7 +164,7 @@ def update(sampled_tensordict):
 
         with timeit("rb - extend"):
             # add to replay buffer
-            tensordict = tensordict.rehsape(-1)
+            tensordict = tensordict.reshape(-1)
             replay_buffer.extend(tensordict.cpu())
         collected_frames += current_frames