mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-01-26 08:29:45 +00:00
26 lines
654 B
Python
26 lines
654 B
Python
from tqdm import tqdm
|
|
import time
|
|
|
|
|
|
class ToolkitProgressBar(tqdm):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.paused = False
|
|
self.last_time = self._time()
|
|
|
|
def pause(self):
|
|
if not self.paused:
|
|
self.paused = True
|
|
self.last_time = self._time()
|
|
|
|
def unpause(self):
|
|
if self.paused:
|
|
self.paused = False
|
|
cur_t = self._time()
|
|
self.start_t += cur_t - self.last_time
|
|
self.last_print_t = cur_t
|
|
|
|
def update(self, *args, **kwargs):
|
|
if not self.paused:
|
|
super().update(*args, **kwargs)
|