Pytorch Lightning
インストール
$conda create -n pytorchlightning
$conda activate pytorchlightning
$ pip install pytorch-lightning or conda install -c conda-forge pytorch-lightning
公式サイト(Github)
https://github.com/PyTorchLightning/pytorch-lightning#how-do-i-do-use-it
Azure machine learningでセットアップした時のサンプル集
https://konchangakita.hatenablog.com/entry/2020/04/25/230000
オフィシャル解説本?
training_step / validation_stepでやっていること
forward()で計算した結果から、損失関数により損失を求め、その値を返す
transformとdatasetのサンプル
Transform, Dataset, Dataloader
Transform:変換器。Datasetがアクセスするデータ群を、ニューラルネットワークのインプットとするために、行う前処理の場所。標準化、次元変換はこちらで。
Dataset:データにアクセスする場所。データベースへの接続やネットからデータのダウンロードなどを行う。
Dataloader:Datasetから実際にデータを取り出す処理。細かく分けて順次取り出すことができる。
参考1 参考2
ログの可視化
ログのパスをちゃんと書いてある事例(Windows)
MSの人の解説、個々の技術は書いてあるが、全体がよくわからない。
サンプルコード一式のリポジトリ(ここを見ろ)
[確認中]trainのデータセットのgetitemは、何をどのような型で返す?
GPUの使い方(Trainer(gpus=xxx))
GPUがあるのに使ってない場合、「UserWarning: GPU available but not used. Set the --gpus flag when calling the script.」のようなログが出力される。GPUが複数コア存在するが、すべて使う場合は、-1を指定する。
例:/home/shuji/Personal/pytorchlightning/basic/pytorchlightning-basic.py --gpus -1
役に立ったら、投げ銭お願いします
ここから先は
¥ 100
この記事が気に入ったらサポートをしてみませんか?