在MXNet框架下,如果要在一个预训练的模型上用你的数据fine-tune一个模型(或者叫迁移学习,即你的模型的参数的初始化不再是随机初始化,而是用别人的在大数据集上训练过的模型的参数来初始化你的模型参数),可以采用MXNet项目自带的fine-tune.py脚本,路径是~/mxnet/example/image-classification/fine-tune.py
MXNet的预训练:fine-tune.py源码详解
猜你喜欢
转载自blog.csdn.net/tony2278/article/details/88304582
今日推荐
周排行