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