可以用torchsummary下的summary方法,或者torchinfo下的summary方法,两者功能是一样的
以torchsummary下的summary方法为例说明
首先通过pip install torchsummary
之后在代码中:
from torchsummary import summary summary(model, (1, 28, 28)) #分别是输入数据的三个维度

可以用torchsummary下的summary方法,或者torchinfo下的summary方法,两者功能是一样的
以torchsummary下的summary方法为例说明
首先通过pip install torchsummary
之后在代码中:
from torchsummary import summary summary(model, (1, 28, 28)) #分别是输入数据的三个维度