caiyuanhao1998 / MST-plus-plus

"MST++: Multi-stage Spectral-wise Transformer for Efficient Spectral Reconstruction" (CVPRW 2022) & (Winner of NTIRE 2022 Spectral Recovery Challenge) and a toolbox for spectral reconstruction

Home Page:https://arxiv.org/abs/2204.07908

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Params和Flops是如何计算的呀

Deep-imagelab opened this issue · comments

请问是直接调用工具包了吗?特别是Flops的计算,是以多大尺寸的输入为基准的呀?

FLOPS 和 Params 需要对网络进行逐层统计。
现有的包如 torchsummaryX 测出来的 Transformer 都不准。

FLOPS 和 Params 需要对网络进行逐层统计。 现有的包如 torchsummaryX 测出来的 Transformer 都不准。

嗯嗯,好的,感谢回复