fix(train): disable tqdm in batch mode to prevent log spam
This commit is contained in:
parent
e81708106d
commit
3c4f0b2ba3
1 changed files with 5 additions and 1 deletions
|
|
@ -1,4 +1,5 @@
|
||||||
import random
|
import random
|
||||||
|
import sys
|
||||||
import time
|
import time
|
||||||
from dataclasses import asdict
|
from dataclasses import asdict
|
||||||
from functools import partial
|
from functools import partial
|
||||||
|
|
@ -244,7 +245,10 @@ def train(args: PPOArgs):
|
||||||
)
|
)
|
||||||
|
|
||||||
print("Starting training...")
|
print("Starting training...")
|
||||||
iters_bar = tqdm.tqdm(range(1, args.num_iterations + 1))
|
iters_bar = tqdm.tqdm(
|
||||||
|
range(1, args.num_iterations + 1),
|
||||||
|
disable=not sys.stdout.isatty(),
|
||||||
|
)
|
||||||
losses = []
|
losses = []
|
||||||
for _ in iters_bar:
|
for _ in iters_bar:
|
||||||
iteration_time_start = time.time()
|
iteration_time_start = time.time()
|
||||||
|
|
|
||||||
Reference in a new issue