torchcv icon indicating copy to clipboard operation
torchcv copied to clipboard
verified publisher icon kuangliu

ssd training cuda problem

Open kelly08385 opened this issue 7 years ago • 0 comments

I have trained my own model on fpnssd. And it can work well. But I tried to run ssd/train.py, the error showed below image It seems like gpu out of memory, but I have trained fpnssd before, there weren't this error. Is fpnssd's parameter less than origin ssd? Or anything I need to modify?

kelly08385 avatar May 17 '18 08:05 kelly08385