make_dot需要一个变量（即带有的张量
grad_fn），而不是模型本身。
尝试：
x = torch.zeros(1, 3, 224, 224, dtype=torch.float, requires_grad=False)out = resnet(x)make_dot(out) # plot graph of variable, not of a nn.Module
尝试：
make_dot需要一个变量（即带有的张量
grad_fn），而不是模型本身。
x = torch.zeros(1, 3, 224, 224, dtype=torch.float, requires_grad=False)out = resnet(x)make_dot(out) # plot graph of variable, not of a nn.Module
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
