pytorch 可视化笔记2:可视化AlexNet
导读:本文共296.5字符,通常情况下阅读需要1分钟。同时您也可以点击右侧朗读,来听本文内容。按键盘←(左) →(右) 方向键可以翻页。
摘要:import torchimport torch.nn as nnimport torchvizimport torchvisionmodel = torchvision.models.AlexNet()x = torch.randn(1,3,224,224)y = model(x)torchviz.make_dot(y, params=dict(model.named_para... ...
目录
(为您整理了一些要点),点击可以直达。import torchimport torch.nn as nnimport torchvizimport torchvisionmodel = torchvision.models.AlexNet()x = torch.randn(1,3,224,224)y = model(x)torchviz.make_dot(y, params=dict(model.named_parameters()))
pytorch 可视化笔记2:可视化AlexNet的详细内容,希望对您有所帮助,信息来源于网络。