TAO 5.0 Classification (PyTorch) deploy error

Refer to Unable to run TAO toolkit in Jetson device - #5 by Morganh