微調

@慕雲發表於2020-11-19

微調

微調過程

步驟 過程
1 源資料集預訓練一個神經網路模型
2 建立新的神經網路模型
3 為目標模型新增一個輸出大小為目標資料集類別個數的輸出層，並隨機初始化該層的模型引數
4 在目標資料集上訓練目標模型

下面是熱狗識別的一個案例
首先，匯入實驗所需的包或模組。Gluon的model_zoo包提供了常用的預訓練模型。GlounCV工具包也可以。

1、獲取資料集

import d2lzh.d2lzh as d2l
from mxnet import gluon,init,nd
from mxnet.gluon  import data as gdata

相關文章