更改预训练模型的默认权重

时间:2020-07-17 06:44:33

标签: tensorflow machine-learning keras deep-learning

我正在研究一个以imagenet权重进行预训练的MobileNet模型,我想以coco权重运行它。如何更改预训练模型的默认权重?

1 个答案:

答案 0 :(得分:0)

当您应用转移学习时,大多数预训练模型都带有一个参数,例如--weights来运行模型,只需寻找它,然后将可可权重作为参数传递