事前学習のやり方(画像分類)
解決したいこと
現在、自作した分類モデルで深層学習による画像分類を行っているのですが
画像枚数の多いデータで事前学習を行いたいのですが方法がわかりません。
大まかなやりかたを教えてください。
フレームワークはpytorchを用いています。
0
現在、自作した分類モデルで深層学習による画像分類を行っているのですが
画像枚数の多いデータで事前学習を行いたいのですが方法がわかりません。
大まかなやりかたを教えてください。
フレームワークはpytorchを用いています。
state_dictをsaveして本学習の際に最初に読み込めばいいのではないでしょうか
@gazounu
Questioner