Error when exporting FrEIA library to ONNX #141
Description
I am trying to export the CFlow-ad model to ONNX, however the export always fails at line 248 in the all_in_one_block.py file.
No error message is displayed, only a graph dump happens.
Any ideas what may be causing this problem?
This is an issue I opened in PyTorch repo.