anibali/dsntnn

Converting to onnx

MetaDev opened this issue · 2 comments

I would like to convert the model to ONNX format but, the operators flip and linspace are not supported. Is there a workaround for this?
Thanks

If you are using DSNT at the output, one option is to save the model up until (but excluding) the DSNT-related operations. Since those layers don't have weights anyway, saving them is not important. Kind of like how you don't need to save the loss function.

This is effectively a duplicate of #9