今天咱们来聊聊用Pytorch的CNN完成kaggle猫狗大战。 话不多说，进入正题。 首先，图片数据来源kaggle，在网站上搜索Dogs vs. Cats很多相关图片集，找一个下载下来。 我这里采用的数据集是： Train：4000张cat + 4000张dog Test：1000张cat + 1000张dog Pytorch版本：(torch 1.3.1+cpu) + (torchvision 0.4.2+cpu) 步骤： 1. 重定义我们的Dataset 2. 定义我们的Pytorch