南理工学渣

摘要： 查看模型每一层的输出情况 以基础的LeNet为例 import torch from torch import nn class Reshape(torch.nn.Module): def forward(self, x): return x.view(-1, 1, 28, 28) net = to 阅读全文