部分源码：
报错如下：
网上有帖子说是多进程的问题，要将 num_works 修改为0，其实个人感觉这个error不管多进程的事。
关于多进程报错 可以参见我另一篇文章：
【pytorch】RuntimeError: Integer division of tensors using div or / is no longer supported【解决】。
这里经过检查修改，我是出错在：torchvision.transforms.Totensor() 没有加括号
加上之后，错误解决。
