model train error
shrutishrestha opened this issue · 1 comments
shrutishrestha commented
I was training this network by my dataset but I got an error. What could be the cause of this?
csrhddlam commented
The cause is CUDA out of memory. You could try reducing the batch size or using a GPU with larger memory.