It seems that torchsummayX does not count torch.nn.parameter layers. import torch import torchsummaryX class Net(torch.nn.Module): def _...
確定! 回上一頁