Python
Java
PHP
IOS
Android
Nodejs
JavaScript
Html5
Windows
Ubuntu
Linux
PyTorch查看网络模型的参数量params和FLOPs等
在PyTorch中 xff0c 可以使用torchstat这个库来查看网络模型的一些信息 xff0c 包括总的参数量params MAdd 显卡内存占用量和FLOPs等 使用前需要先安装torchstat包 xff0c 如下 xff1a p
Pytorch
params
FLOPs
查看网络模型的参数量