PyTorch中的模型分析器-Python开发

torchstat这是基于PyTorch的轻量级神经网络分析仪。 它旨在使您能够快速，轻松地构建网络并进行调试。 注意：此存储库当前正在开发中。 torchstat这是基于PyTorch的轻量级神经网络分析仪。 它旨在使您能够快速，轻松地构建网络并进行调试。 注意：此存储库当前正在开发中。 因此，某些API可能会更改。 该工具可以显示网络参数总数浮点算术（FLOP）的理论数量乘加（MAdd）的理论数量内存使用情况安装有两种方法可以将torchstat安装到您的环境中