Thanks,
Do you mean to use return{loss.to(at::kCPU), acc1},
I think the main difficulty is that the first return argument is tensor while the second return argument is float.
So just wonder if there is a way to return two arguements with different types or I need to convert them to be in the same type.
Thanks.