pytorch

由于自己之前一直在学习tensorflow，后来由于一些原因开始学习pytorch框架了，下面是自己学习一个星期左右写的一个小案例，利用pytorch框架对15中狗狗的类别进行分类，采用resnet50网络，利用不同的方法进行改进，得到如下的准确率曲线，第二种和三种方法类似，但是学习率的不同，由于只训练了5个epoch，准确率还在提升，可以继续训练下去，最后的准确率情况肯定和1一样，虽然他们的方法不同，但作用是一样的。可以看出经过改进，准确率也越来越高。
方案1：

方案2：

方案3：

数据集为15中狗狗的图片：

如下：

用最后的训练权重进行预测，结果如下：