pytorch_memonger icon indicating copy to clipboard operation
pytorch_memonger copied to clipboard

Published 20 hours ago • verified publisher icon prigoyal

benchmark variables are not clear

Open ReyhaneAskari opened this issue 7 years ago • 1 comments

Hi Priya,

I was trying to reproduce your benchmarks results. I think two important information are missing. First, what is the GPU that you used to train these models and how much memory does it have? Second, I don't understand how you have computed max image and what it corresponds to. For example in the VNET, does it correspond to N?

ReyhaneAskari avatar Feb 27 '18 19:02 ReyhaneAskari

Hi @ReyhaneAskari ,

  1. GPU used: M40 with 11443 MiB

  2. Max image is computed by increasing the batch size until the model runs OOM. It will be N for example in VNET

Thanks.

prigoyal avatar Apr 03 '18 14:04 prigoyal