Support for ONNX export

tensorly / torch

TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch

http://tensorly.org/torch/

BSD 3-Clause "New" or "Revised" License

70 stars 18 forks source link

Support for ONNX export #11

Open mfl22 opened 2 years ago

mfl22 commented 2 years ago

Thank you for the library. Have you considered adding support for exporting factorized layers (FactorizedConv) to ONNX?

JeanKossaifi commented 2 years ago

Thanks @mfproto , this is great suggestion. I don't have experience adding layers to ONNX, if you know how to would be awesome if you could start a PR!