Params和Flops是如何计算的呀
Deep-imagelab opened this issue · comments
Deep-imagelab commented
请问是直接调用工具包了吗?特别是Flops的计算,是以多大尺寸的输入为基准的呀?
Yuanhao Cai commented
FLOPS 和 Params 需要对网络进行逐层统计。
现有的包如 torchsummaryX 测出来的 Transformer 都不准。
Deep-imagelab commented
FLOPS 和 Params 需要对网络进行逐层统计。 现有的包如 torchsummaryX 测出来的 Transformer 都不准。
嗯嗯,好的,感谢回复