Closed chi2liu closed 1 year ago
Hi @chi2liu , fx tracing doesn't support control flow, and as a result fx2ait also doesn't support control flow. For now the simplest way is to use Pytorch eager for control flows, and fx2ait for the other part. Thanks.
I use the fx2ait to convert the M2M100 transformer model and it throw a pytorch error for `symbolically traced variables cannot be used as inputs to control flow. I know this is because the control flow is used inside the model. Is there any way for fx2ait to support or bypass this problem?
Original Code:
Error: