adaptive_avg_pool2d does not exist

waleedka / hiddenlayer

Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.

MIT License

1.79k stars 266 forks source link

adaptive_avg_pool2d does not exist #6

Closed ygean closed 5 years ago

ygean commented 5 years ago

Hi, I met this problem when I want to visualize densenet, error shows below: c:\python35\lib\site-packages\torch\onnx\utils.py:446: UserWarning: ONNX export failed on ATen operator adaptive_avg_pool2d because torch.onnx.symbolic.adaptive_avg_pool2d does not exist .format(op_name, op_name)) Actually, there is no adaptive_avg_pool2d in my densenet, only nn.AvgPool2d() exists.

ygean commented 5 years ago

Sorry. It's my mistake that I forget I have add senet module which contains adaptive_avg_pool2d in densenet.

waleedka commented 5 years ago

No worries. Closing this issue then.

John1231983 commented 5 years ago

How about it?

/home/join/anaconda3/lib/python3.6/site-packages/torch/onnx/utils.py:501: UserWarning: ONNX export failed on ATen operator adaptive_avg_pool3d because torch.onnx.symbolic.adaptive_avg_pool3d does not exist