1. 程式人生 > >如何使用預訓練模型對新資料做fine-tune

如何使用預訓練模型對新資料做fine-tune

http://mxnet.io/how_to/finetune.html
使用fine-tune.py檔案對預訓練模型做fine-tune,將最後一層的fc層替換成所需的類別數量和初始化隨機值。例子:從 data/caltech256.sh 下載 caltech256資料集
python fine-tune.py --pretrained-model imagenet11k-resnet-152 --gpus 0,1,2,3,4,5,6,7 \
--data-train data/caltech256-train.rec --data-val data/caltech256-val.rec \
--batch-size 128 --num-classes 256 --num-examples 15240

How to use Fine-tune with Pre-trained Models(

http://mxnet.io/how_to/finetune.html)頁面也有詳細的描述 :

使用resnet-50

使用resnet-152(imagenet-11k)