k/InfiniteMusic
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add status bar for epoch progress

Browse Source
This commit is contained in:
k 2025-11-12 12:12:26 -05:00
parent 579b37cd70
commit b076a0d123
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
train.py
View File

@ -6,6 +6,7 @@ import matplotlib.pyplot as plt
import time
import show
from model import gen
from tqdm import tqdm
BATCH_SIZE = 16
EPOCHS = 100
@ -48,7 +49,7 @@ eshape = (BATCH_SIZE, 1, 128, 216)
for epoch in range(0,EPOCHS):
print(f"\n--- Starting Epoch {epoch} ---\n")
loss=0
for i in range(0,len(x),BATCH_SIZE):
for i in tqdm(range(0,len(x),BATCH_SIZE)):
tx=Tensor(x[i:i+BATCH_SIZE])
ty=Tensor(y[i:i+BATCH_SIZE])
if(tx.shape != eshape):