第三周作业:卷积神经网络(Part1)

Part1

《动手学深度学习》之卷积神经网络

Part2

注:本次使用的数据集并非是AI研习社下载的数据集,因为其数据集太大了,在自己电脑上用cpu跑一次就得两三个小时,所以最后采用的是少量的数据集,也就是pytorch练习的代码中的数据集。

1.AI研习社猫狗大战,首先分析的以前pytorch练习的代码,代码是【 VGG迁移学习进行猫狗大战 】海量高分辨率图像的训练,比较玄学,很难得到一个好的网络。这里我们学习在pretrained VGG网络上 fine-tune,分类猫狗图片。
然后进行简单的调整,效果很好,准确率很高,能达到97.75%。但是并非本周所要求的。

2.通过对原来的代码的学习,进行修改,搭建LeNet网络,进行训练和测试。最终准确率效果实现不好,代码还有待进一步修改。

问题:
1.数据集处理。
2.准确率50%左右。
3.损失出现负值。

注:强化代码能力。

你可能感兴趣的:(人工智能,深度学习,神经网络,深度学习,pytorch)