摘要
Pytork是一个开源的机器学习库,专门研究张量计算、自动微分和GPU加速.出于这些原因,PyTorch是最受欢迎的深度学习的库之一,与Keras和TensorFlow一起广受好评.使用预训练模型的优点一方面是当没有足够的GPU支持时,可以使用别人已经预训练好的模型,如果恰好预训练采用的数据集包含我们自己需要解决的问题的数据,就可以直接使用自己的图片进行预测.另一个优点当然是在训练样本不足的时候,如果采用已经在大规模数据集上训练的模型,可以帮助尽量避免过拟合的发生或者减轻过拟合的程度.本文借助该学习库,进行了一些探索.