使用 torch.load()
載入 my_cnn.pt,再用 model.load_state_dict()
把 my_cnn.pt 的參數導到 model
,再來就可以開始使用 model
了。
下圖為程式碼節錄
把 output
印出來看,會發現有五組數字,每一組數字依序對應到驗證碼圖片的五個文字。
每一組數字裡有 36 個分數,依序代表ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789
每一個字的可能性。
若第一組數字的36的分數中,第 1 個分數最高,代表 model
猜驗證碼第一個文字是 A。
若第一組數字的36的分數中,第 5 個分數最高,代表 model
猜驗證碼第一個文字是 E。
共五組,把五組分數最高的文字取出來,就是 model
猜的驗證碼了。
實驗範例:
🫣
系列文整理:
0. 從零開始,打造第一個 AI 模型 (序言)
1. 環境設定
2. 準備資料集
3. 實做 全連接神經網路 (參數較多)
4. 實做 捲積式神經網路 (參數較少)
5. 訓練、匯出模型
6. 載入、使用模型