@ankkhedia I haven't used specifically VGG19, but I just made a run with Inception-BN and it was able to run with around 3 GB on my laptop GPU (1060) with batch size of 32 on 224X224 images.
Which version did you specifically used? Those exploding RAM explosion should have been fixed with the optimizers refactor. I can try on VGG19 a bit later, but given Inception-BN consumption, I think you might be using mxnet prior to optimizers fix. [ Full content available at: https://github.com/apache/incubator-mxnet/issues/7968 ] This message was relayed via gitbox.apache.org for [email protected]
