HJC256ZY的博客

torchsummary打印模型的每层形状参数import torchimport torchvisionfrom torchsummary import summary #使用 pip install torchsummarydevice = torch.device('cuda' if torch.cuda.is_available() else 'cpu')vgg = torchvision.models.vgg16().to(device)summary(vgg, in