1、使用torchsummary来打印网络特征提取部分参数 实例: importtorch fromtorchsummaryimportsummary frommodelimportAlexNet device=torch.device("cuda"iftorch.cuda.is_available()else"cpu") net=AlexNet().to(device) summary(net,(3,224,224),print=True) 2、使用Netron(onnx形式) (1)使用模型类除了定义一个模型外,还需要传入一个数据样例 importtorchvision.mode...

关注 更多

空空如也 ~ ~

粉丝 更多

空空如也 ~ ~