szq0214/DSOD

Check failed: error == cudaSuccess (2 vs. 0) out of memory.How to deal with it?

githubfeiwang opened this issue · 6 comments

Check failed: error == cudaSuccess (2 vs. 0) out of memory.How to deal with it?

try to reduce the batchsize in DSOD300_pascal.py line 283

@githubfeiwang @szq0214 @wellingtonyl My GPU type is GTX1080 ,only the Train batch_size:4 can run ,Is this normal?

normal, mine is M40, also batch_size=4