torchstat

torchstat 这是一个基于 PyTorch 的轻量级 神经网络 分析器。 它旨在使您快速轻松地构建网络，并具有调试它们的能力。注意：此仓库目前正在开发中。因此，某些 API 可能会发生变化。 此工具可以显示

一个简单的示例

特性与待做事项

注意: 这些特性仅适用于 nn.Module。torch.nn.functional 中的模块尚未受支持。

参考资料