Remove print

dd06205b · sbl1996@126.com · 330ee6af · dd06205b
Commit dd06205b authored May 30, 2024 by sbl1996@126.com
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

scripts/cleanba.py scripts/cleanba.py +0 -1

No files found.
--- a/scripts/cleanba.py
+++ b/scripts/cleanba.py
@@ -916,7 +916,6 @@ def main():
                    b_storage.rewards,
                    b_next_value,
                ))[1]
-            print(jax.tree.map(lambda x: x.shape, (b_storage.dones, target_values, advantages)))

            target_values, advantages = jax.tree.map(
                partial(reshape_batch, num_minibatches=N, num_steps=num_steps),