pip install torchsummary
import torch
import torchvision as tv
from torch.utils.data import DataLoader
import torchvision.transforms as transforms # 必须要有,把PIL.Image.Image变成tensor才能计算
from torchsummary import summary
vgg = tv.models.vgg16(pretrained=True
pytorch可视化之torchsummary
最新推荐文章于 2024-09-17 09:20:50 发布