Aapo Kyrola d37fffd257 use in-place ReLu to safe a lot of memory ...

Summary: Reading Torch docs about Resnets, and soumith's comment,  they mention significant memory-saving with in-place ReLu. prigoyal already had this in her code, but I did not. This saves memory a lot: 9851 MiB -> 7497 MiB.

Reviewed By: prigoyal

Differential Revision: D4346100

fbshipit-source-id: e9c5d5e93787f47487fade668b65b9619bfc9741

2016-12-19 09:29:26 -08:00

..

resnet.py

use in-place ReLu to safe a lot of memory

2016-12-19 09:29:26 -08:00