update torch example#788

cosmicBboy wants to merge 4 commits intomainflyteorg/flyte-sdk:mainfrom

nielsb/update-torch-exampleflyteorg/flyte-sdk:nielsb/update-torch-exampleCopy head branch name to clipboard