remove some unnecessary logging

This commit is contained in:
Noa Aarts 2025-12-06 08:00:17 +01:00
parent 664b5853a5
commit 2b2c81e66e
Signed by: noa
GPG key ID: 1850932741EFF672

View file

@ -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)