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