import json
from PIL import Image
import torch
from torchvision import transforms
from torchviz import make_dot
import hiddenlayer as hl
from efficientnet_pytorch import EfficientNet
model = EfficientNet.from_pretrained('efficientnet-b3')
model.eval()
hl.build_graph(model, torch.zeros([1, 3, 224, 224]))
hl_graph
Traceback (most recent call last):
File "example.py", line 30, in
hl.build_graph(model, torch.zeros([1, 3, 224, 224]))
File "/home/tianhui/hiddenlayer/hiddenlayer/graph.py", line 143, in build_graph
import_graph(g, model, args)
File "/home/tianhui/hiddenlayer/hiddenlayer/pytorch_builder.py", line 71, in import_graph
torch.onnx._optimize_trace(trace, torch.onnx.OperatorExportTypes.ONNX)
File "/home/lanqiang/anaconda3/lib/python3.6/site-packages/torch/onnx/init.py", line 40, in _optimize_trace
trace.set_graph(utils._optimize_graph(trace.graph(), operator_export_type))
File "/home/lanqiang/anaconda3/lib/python3.6/site-packages/torch/onnx/utils.py", line 188, in _optimize_graph
graph = torch._C._jit_pass_onnx(graph, operator_export_type)
File "/home/lanqiang/anaconda3/lib/python3.6/site-packages/torch/onnx/init.py", line 50, in _run_symbolic_function
return utils._run_symbolic_function(*args, *kwargs)
File "/home/lanqiang/anaconda3/lib/python3.6/site-packages/torch/onnx/utils.py", line 589, in _run_symbolic_function
return fn(g, inputs, **attrs)
File "/home/lanqiang/anaconda3/lib/python3.6/site-packages/torch/onnx/symbolic.py", line 130, in wrapper
args = [_parse_arg(arg, arg_desc) for arg, arg_desc in zip(args, arg_descriptors)]
File "/home/lanqiang/anaconda3/lib/python3.6/site-packages/torch/onnx/symbolic.py", line 130, in
args = [_parse_arg(arg, arg_desc) for arg, arg_desc in zip(args, arg_descriptors)]
File "/home/lanqiang/anaconda3/lib/python3.6/site-packages/torch/onnx/symbolic.py", line 90, in _parse_arg
raise RuntimeError("Failed to export an ONNX attribute, "
RuntimeError: Failed to export an ONNX attribute, since it's not constant, please try to make things (e.g., kernel size) static if possible
import json from PIL import Image import torch from torchvision import transforms from torchviz import make_dot import hiddenlayer as hl
from efficientnet_pytorch import EfficientNet model = EfficientNet.from_pretrained('efficientnet-b3') model.eval() hl.build_graph(model, torch.zeros([1, 3, 224, 224])) hl_graph
Traceback (most recent call last): File "example.py", line 30, in
hl.build_graph(model, torch.zeros([1, 3, 224, 224]))
File "/home/tianhui/hiddenlayer/hiddenlayer/graph.py", line 143, in build_graph
import_graph(g, model, args)
File "/home/tianhui/hiddenlayer/hiddenlayer/pytorch_builder.py", line 71, in import_graph
torch.onnx._optimize_trace(trace, torch.onnx.OperatorExportTypes.ONNX)
File "/home/lanqiang/anaconda3/lib/python3.6/site-packages/torch/onnx/init.py", line 40, in _optimize_trace
trace.set_graph(utils._optimize_graph(trace.graph(), operator_export_type))
File "/home/lanqiang/anaconda3/lib/python3.6/site-packages/torch/onnx/utils.py", line 188, in _optimize_graph
graph = torch._C._jit_pass_onnx(graph, operator_export_type)
File "/home/lanqiang/anaconda3/lib/python3.6/site-packages/torch/onnx/init.py", line 50, in _run_symbolic_function
return utils._run_symbolic_function(*args, *kwargs)
File "/home/lanqiang/anaconda3/lib/python3.6/site-packages/torch/onnx/utils.py", line 589, in _run_symbolic_function
return fn(g, inputs, **attrs)
File "/home/lanqiang/anaconda3/lib/python3.6/site-packages/torch/onnx/symbolic.py", line 130, in wrapper
args = [_parse_arg(arg, arg_desc) for arg, arg_desc in zip(args, arg_descriptors)]
File "/home/lanqiang/anaconda3/lib/python3.6/site-packages/torch/onnx/symbolic.py", line 130, in
args = [_parse_arg(arg, arg_desc) for arg, arg_desc in zip(args, arg_descriptors)]
File "/home/lanqiang/anaconda3/lib/python3.6/site-packages/torch/onnx/symbolic.py", line 90, in _parse_arg
raise RuntimeError("Failed to export an ONNX attribute, "
RuntimeError: Failed to export an ONNX attribute, since it's not constant, please try to make things (e.g., kernel size) static if possible