使用calflops计算网络的模型参数和FLOPs

from calflops import calculate_flops
model = models.resnet18()
flops, macs, params =calculate_flops(model=model,
input_shape(1,3,224,224),output_as_string=True,output_precision=4)

如果出现ModuleNotFoundError: No module named 'transformers

请直接 pip install transformer

上面的不是resnet18的参数！ 是我随机编的。