search

alexlee-gk / lpips-tensorflow

Tensorflow port for the Learned Perceptual Image Patch Similarity (LPIPS) metric.
BSD 2-Clause "Simplified" License
130 stars 30 forks source link

Unable to export models to tf: "NotImplementedError: ATen is not implemented." #7

Open yash0307 opened 4 years ago

yash0307 commented 4 years ago

I've the recommended version of torch, torchvision and latest versions of onxx. However, I am unable to export the model since Aten is not supported in onxx. How did you export them ?

Lack of Aten support is a known issue, however, I couldn't find an easy work-around. Please help.

yash0307 commented 4 years ago

Note that this only happens for "--model net"