[GitHub] anirudhacharya commented on issue #13960: Use In-place operator to prevent memory spikes in optimizer updates

[GitBox]

anirudhacharya commented on issue #13960: Use In-place operator to prevent 
memory spikes in optimizer updates
URL: https://github.com/apache/incubator-mxnet/pull/13960#issuecomment-457492030
 
 
   For 5 batches of training with a deep embeddings example the improvement in 
memory consumption is ~3.6 factor for Nesterov Adam optimizer.
   
   **Old Nadam -**
   
   <img width="1440" alt="old_nadam" 
src="https://user-images.githubusercontent.com/2778341/51733273-ad0a4e00-2035-11e9-9314-01cee059d9b3.png";>
   
   **New Nadam( with in-place operators) -** 
   
   <img width="1440" alt="new_nadam" 
src="https://user-images.githubusercontent.com/2778341/51733284-b5fb1f80-2035-11e9-9198-5ad37d040e5b.png";>
   

----------------------------------------------------------------
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:
us...@infra.apache.org


With regards,
Apache Git Services