Add --cdc_debug flag and tqdm progress for CDC preprocessing

- Add --cdc_debug flag to enable verbose bucket-by-bucket output
- When debug=False (default): Show tqdm progress bar, concise logging
- When debug=True: Show detailed bucket information, no progress bar
- Improves user experience during CDC cache generation
This commit is contained in:
rockerBOO
2025-10-09 17:24:02 -04:00
parent 7a7110cdc6
commit c8a4e99074
4 changed files with 38 additions and 19 deletions

View File

@@ -635,6 +635,7 @@ class NetworkTrainer:
gamma=args.cdc_gamma,
force_recache=args.force_recache_cdc,
accelerator=accelerator,
debug=getattr(args, 'cdc_debug', False),
)
else:
self.cdc_cache_path = None