diff --git a/sota-implementations/cql/cql_offline.py b/sota-implementations/cql/cql_offline.py index 99b391b9db8..441cb3555e2 100644 --- 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 index da65c5c246e..66c6d206c3d 100644 --- 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()