torch-summary只能看到网络结构和参数数量: torchstat 可以看到更多的信息,如参数的内存大小: 关于计算指标: FLOPS、MAdds、MACC指标:
m.total_ops = torch.Tensor([int(total_ops)]) 总体而言,模型会计算每一个卷积核发生的乘加运算数,再推广到整个卷积层级的总乘加运算数。 定制你的运算统计 有一些运算统计还没有加进去,如果我们知道该怎样算,那么就可以写个自定义函数。 classYourModule(nn.Module):# your definitiondefcount_your_model(...