precompute training data outside training script
This commit is contained in:
1
.gitignore
vendored
1
.gitignore
vendored
@@ -2,3 +2,4 @@
|
|||||||
/__pycache__/
|
/__pycache__/
|
||||||
/data/
|
/data/
|
||||||
/music.safetensors
|
/music.safetensors
|
||||||
|
/data.npz
|
||||||
|
|||||||
13
dataInit.py
Normal file
13
dataInit.py
Normal 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)
|
||||||
Reference in New Issue
Block a user