Reset CUDA memory cache after first step (reduces chance of OOM)
Reset CUDA memory cache after first step (reduces chance of OOM)

Summary: Pull Request resolved: https://github.com/fairinternal/fairseq-py/pull/1001

Differential Revision: D19471949

Pulled By: myleott

fbshipit-source-id: f929aa5b0f3da37abf03ef1b5844bd2637366f5e

pytorch/fairseq,pytorch/fairseq,pytorch/fairseq