pytorch模型可视化torchviz

阿里云2000元红包!本站用户参与享受九折优惠!

最好用jnotebook

!pip install torchviz

import torch

from torch import nn

from torchviz import make_dot, make_dot_from_trace

from torchvision import models

model = models.vgg19()

x = torch.randn(10,3,224,224)

make_dot(model(x), params=dict(model.named_parameters()))

效果:虽然有点看不懂,但是聊胜于无。


https://www.jianshu.com/p/54edab88d76e

「点点赞赏,手留余香」

    还没有人赞赏,快来当第一个赞赏的人吧!
0 条回复 A 作者 M 管理员
    所有的伟大,都源于一个勇敢的开始!
欢迎您,新朋友,感谢参与互动!欢迎您 {{author}},您在本站有{{commentsCount}}条评论