まだタイトルない

アウトプット用です

2021-08-26から1日間の記事一覧

【PyTorch Lightning】checkpointをロードするのに躓いたことメモ

こんにちは 最近PyTorch Lightningで学習をし始めてcallbackなどの活用で任意の時点でのチェックポイントを保存できるようになりました。 save_weights_only=Trueと設定したの今まで通りpure pythonで学習済み重みをLoadして推論できると思っていたのですが…