torchsnooper专题

pytorch实用工具:torchsummary、torchsnooper

一. torchsummmary工具: 其主要是用来计算网络的计算参数等信息的,其大概的流程为: 安装指令: pip install torchsummary 使用方法: import torch, torchvisionmodel = torchvision.models.vggmodel = torchvision.models.vgg16()from torchsummary