pytorch 可视化笔记2:可视化AlexNet(教程,深度学习,编程语言)

时间:2023-9-24 作者 : 石家庄SEO 分类 : 编程语言
  • TAG :
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的详细内容,希望对您有所帮助,信息来源于网络。
上一篇:Valine评论之Valine-admin配置攻略下一篇:

3 人围观 / 0 条评论 ↓快速评论↓