diff --git a/blokus.py b/blokus.py index 53b4b1e..ba7d533 100755 --- a/blokus.py +++ b/blokus.py @@ -417,10 +417,7 @@ def train( # Update progress bar with most recent stats pbar.set_postfix( - episode=episode, loss=loss_value, - r1=float(r1), - r2=float(r2), ) # Save checkpoint every N episodes (and at the very end)