请问论文里的FLOPs是用什么函数或者库计算的呀

MrBlankness / LightM-UNet

Pytorch implementation of "LightM-UNet: Mamba Assists in Lightweight UNet for Medical Image Segmentation"

https://arxiv.org/abs/2403.05246

Apache License 2.0

279 stars 25 forks source link

Closed LIU-YUXI closed 6 months ago

LIU-YUXI commented 7 months ago

请问论文里的FLOPs是用什么函数或者库计算的呀，是只计算了前向传播的FLOPs吗？谢谢~

MrBlankness commented 6 months ago

请问论文里的FLOPs是用什么函数或者库计算的呀，是只计算了前向传播的FLOPs吗？谢谢~

We use the thop library to compute the FLOPs of the model. Yes, we only calculate the FLOPs for the forward pass.

LIU-YUXI commented 6 months ago

好的非常感谢！