How to measure the memory consumption of model?

Zheng222 / IMDN

Lightweight Image Super-Resolution with Information Multi-distillation Network (ACM MM 2019, Winner Award of ICCVW AIM 2019 Constrained SR Track1&Track2)

418 stars 69 forks source link

How to measure the memory consumption of model? #3

Closed minghongli233 closed 4 years ago

minghongli233 commented 4 years ago

I want to know how to measure the memory consumption of model in your paper. Thank you!

Zheng222 commented 4 years ago

@minghongli233 We simply record the memory consumption by execute watch -n0.1 nvidia-smi when testing the model.