如何訓練預訓練 model (MNIST) tensorflow

Question

我有一個 Fashion MNIST 項目，它根據上傳的圖像預測衣服，我想用它做一些改進。 是否可以修改我的項目，使其在每次上傳和預測后自動訓練？

Answer 1

您可以使用遷移學習技術手動訓練您的 model（遷移學習是一種將已訓練的 model 重新用於另一項任務的方法）。

1. 實例化一個基礎 model 並將預訓練的權重加載到其中。
2. 通過設置 trainable = False 凍結基礎 model 中的所有層。
3. 在基礎 model 的一個（或多個）層的 output 之上創建一個新的 model。在新數據集上訓練新的 model。

請參閱此要點以獲取工作代碼示例。 謝謝你。