microsoft / LQ-Nets

LQ-Nets: Learned Quantization for Highly Accurate and Compact Deep Neural Networks
MIT License
239 stars 70 forks source link

what is the size needed for LQ-Nets vgg model to run #12

Closed monkeyking closed 5 years ago

monkeyking commented 5 years ago

When I tried to run the VGG model ,I see the RAM quickly ran out and LQ-Nets be killed by system. I am not sure my ram is enough for the Net to run.

monkeyking commented 5 years ago

system : ubuntu 18 ram : 16GB gpu : 1080TI

EowinYe commented 5 years ago

How much is the ram in 1080TI? I ran it on M40 with 24G before. Since VGG needs very large memories, you can try to decrease the batch size if you'd like to run it on 1080TI (Also decrease the learning rate).