I’m glad you figured it out!
During evaluation, i.e. after you call model.eval(), you won’t get the auxiliary output.
So you just have to pass outputs[0] while training and can pass outputs directly during evaluation.