Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-09-20 18:45:09 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

add total time estimate

Browse source
This commit is contained in:
Alex "mcmonkey" Goodwin 2023-03-27 10:57:27 -07:00
parent 16ea4fc36d
commit 9ced75746d
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
modules/training.py
View file

@ -188,13 +188,15 @@ def do_train(loraName: str, microBatchSize: int, batchSize: int, epochs: int, le
timeElapsed = time.perf_counter() - startTime timeElapsed = time.perf_counter() - startTime
if timeElapsed <= 0: if timeElapsed <= 0:
timerInfo = "" timerInfo = ""
totalTimeEstimate = 999
else: else:
its = CURRENT_STEPS / timeElapsed its = CURRENT_STEPS / timeElapsed
if its > 1: if its > 1:
timerInfo = f"`{its:.2f}` it/s" timerInfo = f"`{its:.2f}` it/s"
else: else:
timerInfo = f"`{1.0/its:.2f}` s/it" timerInfo = f"`{1.0/its:.2f}` s/it"
yield f"Running... **{CURRENT_STEPS}** / **{MAX_STEPS}** ... {timerInfo}, `{timeElapsed:.1f}` seconds" totalTimeEstimate = (1.0/its) * (MAX_STEPS)
yield f"Running... **{CURRENT_STEPS}** / **{MAX_STEPS}** ... {timerInfo}, `{timeElapsed:.0f}`/`{totalTimeEstimate:.0f}` seconds"
print("Training complete, saving...") print("Training complete, saving...")
loraModel.save_pretrained(loraName) loraModel.save_pretrained(loraName)
if WANT_INTERRUPT: if WANT_INTERRUPT: