pytorch summary:PyTorch中的模型摘要类似于Keras中的`model.summary()`。 源码

model.summary()中的Keras样式model.summary() Keras有一个简洁的API来查看模型的可视化,这在调试网络时非常有帮助。 这是一个尝试在PyTorch中模仿的准系统代码。 目的是提供与print(your_model)中的print(your_model)无法提供的信息互补的信息。 用法 pip install torchsummary或 git clone https://github.com/sksq96/pytorch-summary from torchsummary import summary summary ( your_model , in