可以用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)) #分别是输入数据的三个维度