finetune model

[shuangshuangguo]

Hi, @dalgu90
Thanks for your wonderful work! Could you share your fine-tuned model based on the transferred model? I have no devices to fine-tune the model on ImageNet, but I want to use the ResNet18 model to fine-tune my own dataset.

[prerit2010]

@dalgu90 what do we change in extract_torch_t7.py to use the pretrained model to train on my dataset of 5 classes ?