From 5ffd212a67d6f095eea36528d49a684debc79c33 Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Sun, 15 Dec 2024 10:24:37 -0800 Subject: [PATCH] Update [ghstack-poisoned] --- sota-implementations/iql/iql_online.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sota-implementations/iql/iql_online.py b/sota-implementations/iql/iql_online.py index 499c2164b52..6e9c8a0d8ea 100644 --- 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