import torch
from torchvision import models
from torchviz import make_dot
model = models.resnet50()
x = torch.randn(1, 3, 224, 224)
vis_graph = make_dot(model(x),params=dict(model.named_parameters()))
vise_graph.view()
import torch
from torchvision import models
from torchviz import make_dot
model = models.resnet50()
x = torch.randn(1, 3, 224, 224)
vis_graph = make_dot(model(x),params=dict(model.named_parameters()))
vise_graph.view()