from torchsummary import summary
summary(net, input_size=(3, 256, 256), batch_size=-1)
输出的参数是除以一百万(/1000000)M,
from fvcore.nn import FlopCountAnalysis
inputs
from torchsummary import summary
summary(net, input_size=(3, 256, 256), batch_size=-1)
输出的参数是除以一百万(/1000000)M,
from fvcore.nn import FlopCountAnalysis
inputs
1万+
2万+

被折叠的 条评论
为什么被折叠?