I see why I got a tensor output. I set the model to eval mode.
The source file is pretty clear, I assume you have figured out what the output tuple contains are and how to disable outputting tuple.
Thanks.