使用基本测试代码表单隐藏层，我得到的错误标题：
import torch
import torchvision.models
import hiddenlayer as hl
# VGG16 with BatchNorm
model = torchvision.models.vgg16()
# Build HiddenLayer graph
# Jupyter Notebook renders it automatically
hl.build_graph(model, torch.zeros([1, 3, 224, 224]))版本：隐藏层-0.3 pytorch=1.13.0+ cu 117 Python=3.10.6
1条答案
按热度按时间e4yzc0pl1#
我按照错误建议将pytorch_builder.py第71行中的_optimize_trace更改为_optimize_graph。之后，它正常工作。