[BugFix] Fix CQL/IQL pbar update (#2020)

pytorch · Mar 18, 2024 · 9747170 · 9747170
1 parent c29c345
commit 9747170
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sota-implementations/cql/cql_offline.py b/sota-implementations/cql/cql_offline.py
@@ -81,7 +81,7 @@ def main(cfg: "DictConfig"):  # noqa: F821
     # Training loop
     start_time = time.time()
     for i in range(gradient_steps):
-        pbar.update(i)
+        pbar.update(1)
         # sample data
         data = replay_buffer.sample()
         # compute loss

diff --git a/sota-implementations/iql/iql_offline.py b/sota-implementations/iql/iql_offline.py
@@ -82,7 +82,7 @@ def main(cfg: "DictConfig"):  # noqa: F821
     # Training loop
     start_time = time.time()
     for i in range(gradient_steps):
-        pbar.update(i)
+        pbar.update(1)
         # sample data
         data = replay_buffer.sample()