szha commented on issue #9852: Custom Function Massive Memory Leak URL: https://github.com/apache/incubator-mxnet/issues/9852#issuecomment-367443424 This is because the forward loop doesn't wait for all computation to finish. ndarray is executed asynchronously. To address it, add `nd.waitall()` after `out = net(test)`. Usually when you need to monitor the training/testing accuracy and performance of models, you'd call `array.asnumpy()`, which is a blocking function, so it's not necessary to manually add additional `nd.waitall()`
---------------------------------------------------------------- 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
