STANet
STANet copied to clipboard
关于模型参数量
感谢您的工作。 请问 BAM 和 PAM 的模型参数量和 FLOPs 情况如何呢?
多谢关注哈。我们没有统计过模型参数量和FLOPs的情况,您感兴趣可以测一测,可以参考开源工具:https://github.com/Lyken17/pytorch-OpCounter
你好@likyoo,请问一下您有计算参数量和 FLOPs 吗?该如何使用thop计算双输入?