Solved by ptrblck in post #4 
              
              
                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?