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的参数! 是我随机编的。