PyTorch

Pytorchで学習したモデルを保存する時は

torch.save(model.state_dict(), model_path)

でモデルを保存することが推奨される．

state_dictがsaveで直接保存す ...

Pytorchを使っている時に出たエラー．これがでたときには，モデルの読み込みの順番でエラーが発生していることが多い．少し順番を変えてあげればケロっとエラーが解消されることが多い．

よくあるエラーが，.load_state ...

import osimport globimport randomimport numpy as npimport pandas as pdfrom sklearn.model_selection import train_test_spl ...

from glob import globfrom PIL import Imageimport numpy as npimport torchimport torch.utils.data as data_utilsfrom torchv ...

Pytorchのコードを見ているとミニバッチごとのlossやaccuracyを計算する際、.dataを用いて値を取り出されることが頻繁にある。

よくある例:

for i in range(0, 2 * POS_NEG ...

import torchprint(torch.cuda.is_available())>>Truetorch.cuda.get_device_name(0)>>'GeForce GTX 1080 Ti'

Pytorchに関する基礎知識 ...