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

precompute training data outside training script

Browse source
This commit is contained in:
k 2025-10-15 02:23:35 -04:00
parent 78ece6996f
commit c8c6ae03ac
2 changed files with 14 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
.gitignore vendored
View file

@ -2,3 +2,4 @@
/__pycache__/ /__pycache__/
/data/ /data/
/music.safetensors /music.safetensors
/data.npz

13
dataInit.py Normal file
View file

@ -0,0 +1,13 @@
import data
import numpy as np
x,y = data.dataset(data.load())
size=len(x)
x_np = np.stack(x)
x_np = np.expand_dims(x_np, axis=1)
y_np = np.stack(y)
y_np = np.expand_dims(y_np, axis=1)
np.savez_compressed("data",x_np,y_np)