PyTorch 23 1月 2021 torch.saveとstate_dictの違い Pytorchで学習したモデルを保存する時は torch.save(model.state_dict(), model_path) でモデルを保存することが推奨される. state_dictがsave[…] 続きを読む
Python 23 1月 2021 AttributeError: ‘_IncompatibleKeys’ object has no attribute ‘to’ Pytorchを使っている時に出たエラー.これがでたときには,モデルの読み込みの順番でエラーが発生していることが多い.少し順番を変えてあげればケロっとエラーが解消されることが多い. よくあるエラーが,[…] 続きを読む
Python 21 6月 2020 VGG16 pre-trainedの中間層を用いたモデル構築例[pytorch] import os import glob import random import numpy as np import pandas as pd from sklearn.model_select[…] 続きを読む
PyTorch 2 1月 2019 TensorDatasetで生画像をTensor化して読み込む[PyTorch] from glob import glob from PIL import Image import numpy as np import torch import torch.utils.data […] 続きを読む
PyTorch 11 10月 2018 Pytorch v0.4のコードをv0.3で動かす際には.dataに注意[Pytorch] Pytorchのコードを見ているとミニバッチごとのlossやaccuracyを計算する際、.dataを用いて値を取り出されることが頻繁にある。 よくある例: for i in range(0, 2 *[…] 続きを読む
DNN 30 8月 2018 PytorchがGPUを使っているかを確認する方法[Pytorch] import torch print(torch.cuda.is_available()) >> True torch.cuda.get_device_name(0) >> 'GeForce GTX […] 続きを読む