pytorch图像分类器运行报错的解决方案

问题

报错：

BrokenPipeError: [Errno 32] Broken pipe

RuntimeError: bool value of Tensor with more than one value is ambiguous

可能的解决方案：

1. 用def定义了函数，却没加 if __name__ =="__main__":。
   将除了函数和变量之外的东西全部扔到main中。
2. 声明loss函数的时候记得加上括号。（没有参数设置容易漏掉括号）
   criterion = nn.CrossEntropyLoss()#记得加括号！！！