pytorch example study-training a classifier www.programmerall.com