Pythonいぬ

pythonを使った画像処理に関する記事を書いていきます

2020-12-01から1ヶ月間の記事一覧

Pytorchで学習のチェックポイントの保存

Pytorchでモデルを保存する場合、モデルのパラメータのみを保存することが多い。しかし、モデルパラメータだけではlossがどれくらいか、optimizerは何を使ったか、何イテレーション学習してあるかなどの情報がわからない。これらがわからないと特に途中から…