louisYen/S3R

MemoryError: Unable to allocate 4.66 GiB for an array with shape (61032, 10, 2048) and data type float32

Opened this issue · 0 comments

Thanks for your outstanding contribution
I also train the model on RTX2080Ti
but in Ubuntu 16.04
and I have a problem with memory
MemoryError: Unable to allocate 4.66 GiB for an array with shape (61032, 10, 2048) and data type float32
I have tried to modify the batch size from 32 to 16, 8. And still cannot fix the problem
and this is my result
batch size 32 run 4/15000epoch
batch size 16 run 12/15000epoch
batch size 8 run 25/15000epoch
and all of above are broken
how can i fix it