Want to know how much VRAM you will need for training your model?
Now you can use this webapp in which you can input a torch/tensorflow summary or the parameters count and get an estimate of the required memory!
The url is: howmuchvram.com
Also, everything is Open Source so you can contribute in repo!
Leave a star if you find it useful!