Pytorch

1/2ページ

Pytorch Lightning – 事前学習モデルを使ってクラス分類モデルを学習する方法

画像のクラス分類問題の学習を Pytorch Lightning を使用して行う方法について解説します。Pytorch で行う場合のコードは以下の記事で解説していますが、Pytorch Lightning を使用することで Pytorch の冗長なコードを大幅に減らすことができ、コードの見通しがよくなります。{url=pytorch-train-classification-problem-using-a-pretrained-model}