pytorch训练代码,就是这样啦,应该是实现imagenet的分类器吧