fix memory_used_ by computing after SetUp
Merge pull request #980 from jeffdonahue/fix-memory-used

fix memory_used_ by computing after SetUp