From 14beba63391e1ddc9a145bb638d9306086ad1a5c Mon Sep 17 00:00:00 2001 From: Volpeon Date: Fri, 7 Oct 2022 08:34:07 +0200 Subject: Training: Create multiple class images per training image --- infer.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'infer.py') diff --git a/infer.py b/infer.py index 34e570a..6197aa3 100644 --- a/infer.py +++ b/infer.py @@ -262,7 +262,10 @@ def generate(output_dir, pipeline, args): with autocast("cuda"): for i in range(args.batch_num): - pipeline.set_progress_bar_config(desc=f"Batch {i + 1} of {args.batch_num}") + pipeline.set_progress_bar_config( + desc=f"Batch {i + 1} of {args.batch_num}", + dynamic_ncols=True + ) generator = torch.Generator(device="cuda").manual_seed(seed + i) images = pipeline( -- cgit v1.2.3-54-g00ecf