speedinghzl / CCNet

CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).
MIT License
1.42k stars 277 forks source link

Memory usage #77

Open songqi-github opened 4 years ago

songqi-github commented 4 years ago

How to compute the Memory usage of RCCA? Could you please share the details? Thanks a lot !

speedinghzl commented 4 years ago

The memory usage of RCCA could be obtained by the memory usage of CCNet minus the memory usage of baseline (CCNet without RCCA module). The nvidia-smi tool is used to monitor GPU memory usage.