- 博客(1)
- 收藏
- 关注
原创 Pytorch模型的网络结构可视化
import torchfrom torchvision import modelsfrom torchviz import make_dotmodel = models.resnet50()x = torch.randn(1, 3, 224, 224)vis_graph = make_dot(model(x),params=dict(model.named_parameter...
2019-06-28 16:13:33 4195
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人