pytorch でハマりやすいポイント - ryuiso2のブログtitle pytorch (pytorch=1.1.0) 基本的にはレファレンスを参照するが、躓きやすいところを中心にメモを残す。 GPUの使用 ソースコード中に明示的にGPU用変数に設定する必要がある。 モデル、変数、計算に関わるものすべてを .to(device) しなければならない。 device = 'cuda' CNN().to(device) # 以下のように書くと便利 import torch device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') 正しくGPUが使用できているかの確認は nvidia-smi -l でGPUの使用率を確認する。 Parameter の作成 Parameter とは、学習によって調整する学習パラメータのことである。 括弧 () の順番が大切で[dim1, di
