Rework progress bar and ui speed string so it gets the same value at a more accurate and snappier wall clock time

This commit is contained in:
Jaret Burkett
2026-08-21 13:31:02 -06:00
parent b96476a841
commit 27a03a91f2
3 changed files with 17 additions and 19 deletions

View File

@@ -4,6 +4,9 @@ import time
class ToolkitProgressBar(tqdm):
def __init__(self, *args, **kwargs):
# high EMA alpha so the rate/ETA responds within a few steps
# (tqdm default of 0.3 takes ~10 steps to wash out old samples)
kwargs.setdefault('smoothing', 0.7)
super().__init__(*args, **kwargs)
self.paused = False
self.last_time = self._time()