如何在pytorch中使用自己训练有素的模型进行迁移学习?

时间:2019-06-07 12:39:15

标签: pytorch transfer-learning

使用转移学习但没有在线预训练模型,我不想使用Torchvision.models和其他任何在线可用的方法。我有一个在pytorch中从头开始训练的模型,现在我希望它可以在其他对象上传输以进行图像分类。请注意,在这种新情况下,标签的数量是不同的。

从头开始在pytorch中训练特定数据集上的模型。但现在我想将其传输到新的数据集(更改的类数)

0 个答案:

没有答案