3 ms·
This question might be a bit off topic but does anyone know of a calculator or library (I primarily use Tensorflow and Keras) to calculate the GPU memory footpr
by syntaxing 10y ago
This question might be a bit off topic but does anyone know of a calculator or library (I primarily use Tensorflow and Keras) to calculate the GPU memory footprint of the network? I only have a 1050Ti and I fit my networks in the memory by manually adjusting different parameters. It would be nice to know how much room I have left to expand my network.
- Q6T46nT668w6i3m 10y agomodel.summary() in Keras will provide you the number of parameters. You can use this with the shape of your data to measure the memory cost.
- syntaxing 10y agoThanks! Can you explain how I can calculate the memory cost with the shape of my data assuming a batch size of 1 and a input data shape of (10,20,3)?
- deleted 10y ago[deleted]
- deleted 10y ago[deleted]