Hi, I am trying to export the segmentation some segementation of the segmentation model, newly added to torchvision. They export sucessfully after applying this workaround 14. However the ONNX graph generates several cast operations as seen in the picture.

image.png1919×303 92.9 KB

It says its coming from the ASPPPooling layer but that layer is defined as:

image.png719×168 23.6 KB

And I dont think any of these layers do any casting. Any ideas where this Cast[to=1] comes from or how to avoid it? Any help will be greatly appreciated!! 
P.S I am using pytorch and torchvision both compiled from source