transfer_learning_demo 源码

作者:杨坦 该项目是在CIFAR-100数据集上对ImageNet预训练模型的头部分类器进行再训练的演示。 下载预训练模型:[ ]并将其另存为pretrained_model/resnet18-5c106cde.pth 在“脚本”文件夹中,您可以直接运行“ train.sh”来训练模型。 数据集将自动下载。 有关更多选项,请参阅“ train_head_classifier.py”。