pytorch可视化输出模型计算力和参数量，设备损耗，运行时间等信息

输出下面的表格形式的，将自己的模型都输出出来。

from torchstat import stat
from torchvision.models import resnet50
model = resnet50()
stat(model, (3, 224, 224))  #  (3,224,224)表示输入图片的尺寸

记得：

pip install torchstat

不过好像会报下面的错误：

所以，只需要在最下面的错，点进去了。将detach()后面变为detach().cpu,bu，不过还是会报下面的错误。这个提示是版本问题，解决不了，出了让所有的版本对应，但是成本太大，放弃！
可以参考这这篇文章：https://blog.csdn.net/qq_44666320/article/details/106093613

计算运行时间：

torch.cuda.synchronize()
start = time.time()
XXX（进程）
XXX（进程）
XXX(进程)
torch.cuda.synchronize()
end = time.time()
total_time = end - start
print('total_time:{:.2f}'.format(total_time))