lichen11 commented on issue #7968: [R] Transfer Learning using VGG-16 URL: https://github.com/apache/incubator-mxnet/issues/7968#issuecomment-355711093 I added gc() before mx.model.FeedForward.create and also within mx.model.FeedForward.create. Then I immediately ran nvidia-smi. At some second, utilization was 29%, then GPU Memory Usage reached to about 12.7GB for a few seconds, then R crashes.
---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected] With regards, Apache Git Services
