prigoyal / pytorch_memonger

Experimental ground for optimizing memory of pytorch models

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Benchmark clarification

dmitrivainbrand opened this issue · comments

Hi Priya,
It is unclear how you run the benchmark. I noticed that for best results I need to run each unit-test independently. Otherwise the first tests seem to "steal" memory from later tests. If the benchmark is intended to run as a whole, probably need to find a way to clean the memory between the tests. Otherwise, maybe need to state clearly in the Readme that the tests are supposed to run independently.
Thanks
Dmitri