torchkeras专题

使用torchkeras打印Pytorch模型结构和基本参数信息

在使用Pytorch构建神经网络模型后,我们需要看一下自己写的模型的网络结构,此时可以使用torchkeras模块中的summary函数实现该功能。以多层感知机为例,首先我们构建网络结构并打印该模型的初步信息,代码如下: import torchfrom torch import nnfrom torchkeras import summarydef create_net():net = n