refactor: use tqdm instead of custom load bar

This commit is contained in:
Lenoctambule
2026-03-27 22:16:17 +01:00
parent 32e27b4b62
commit c37d1c9c26
2 changed files with 30 additions and 26 deletions

View File

@@ -4,6 +4,7 @@ from utils import (regularize,
dynamic_loss_plot_update,
dynamic_loss_plot_finish)
import types
from tqdm import tqdm
LOADER = ['', '', '', '', '', '', '', '']
@@ -88,10 +89,12 @@ class Autoencoder:
epoch = 0
no_improv = 0
prev_error = float('inf')
with tqdm(bar_format="{desc} {elapsed} {rate_fmt}") as lbar :
while True:
print(
f"{LOADER[epoch % len(LOADER)]} Training \t({epoch=} error={prev_error:.2f})", # noqa
lbar.set_description(
f"{LOADER[epoch % len(LOADER)]} Training ({epoch=} error={prev_error:.2f})",
)
lbar.update()
error = 0
for x in data_set:
input = x.flatten()

View File

@@ -1,3 +1,4 @@
numpy
matplotlib
requests
tqdm