szagoruyko / wide-residual-networks

3.8% and 18.3% on CIFAR-10 and CIFAR-100
http://arxiv.org/abs/1605.07146
BSD 2-Clause "Simplified" License
1.3k stars 293 forks source link

image size? tried 224x224, but out of GPU memory on 12G Titan X #43

Open mw66 opened 6 years ago

mw66 commented 6 years ago

looks like WRN need too much GPU memory, I tried image size of 224x224, and even with depth 10 (1x6 + 4), it run out of GPU memory.

I wonder can the memory can be reduced?

Thanks.