2.第二部加载数据集，定义模型，进行训练，训练过之后会保存一个权重文件，你可以加载这个权重文件进行再训练，迁移学习。3.训练出权重文件，开始进行预测（需要导入Net或者再写一遍模型），out的输出是数据集中的类别序号。1.首先是写一个nn。