vbelz / Speech-enhancement

Deep learning for audio denoising
MIT License
646 stars 124 forks source link

MemoryError #31

Open Titook opened 1 year ago

Titook commented 1 year ago

Hello blogger, I encountered an error: MemoryError: Unable to allocate array with shape (20000, 128, 128) and data, where I changed 40000 to 20000, but there is still this issue. I would like to ask if this is due to excessive training set data or NB_ The samples are too large

Titook commented 1 year ago

At the same time, I also have a question that your training model does not define whether to use CPU or GPU. Is this the default CPU for training, and how should I change it to GPU for training? Thank you.