MemoryError

[Titook]

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]

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.