hiddenlayer
hiddenlayer copied to clipboard
Failed to export an ONNX attribute, since it's not constant, please try to make things (e.g., kernel size) static if possible
Hi, I meet the problem using torchvision model = torchvision.models.segmentation.deeplabv3_resnet101(pretrained=True) model.eval() trace, _ = torch.jit.get_trace_graph(model, torch.randn(1, 3, 224, 224)) torch.onnx._optimize_trace(trace, torch.onnx.OperatorExportTypes.ONNX) Could you please give some hits to save this problem?
Same problem here, trying to visualize a model from https://github.com/JiaminRen/RandWireNN
Hello I also faced the same issue, did you guys managed to solve it ?