3 ms·
model.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.
by Q6T46nT668w6i3m 10y ago
model.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)?