crook52:

Is it correct to understand PyTorch eager mode as normal mode?


Yes, sorry for the unclear naming. By “eager” mode I meant the normal Python usage.
Good to see the model is working generally.
Could you , for the sake of debugging, remove the logits_aux from the forward and just return the logits and retry to export the model?