tmori’s blog

公開メモ帳くらいの感覚で書いています。技術系多め。日常少なめ。

PyTorch

【Python】オフラインでfine-tuningのpretrainedを使用する方法【PyTorch】

ネットワーク環境にないGPU計算機でfine-tuningのpretrainedをする方法についてまとめます。 今回はresnet18を使用するケースを考えていますが、同じ手順で他のモデルも同様にオフライン実行できると思います。 目次 目次 環境 fine-tuningのpretrainedとは …

【Python】[Errno 32] Broken pipe【PyTroch】

pytorchで [Errno 32] Broken pipe の解消方法について 実行環境 windows 10 PyTorch v1.2 エラー発生原因 このエラーはwindows+torch.utils.data.DataLoaderのnum_workers>1 + IPythonの書き方で発生してるようです。 同じコードをcentosで実行すると正常…

RuntimeError: cuda runtime error(59) : device-side assert triggered で詰まった話

pytorchでMLP使って多クラス分類したときのエラーについて書きます。 環境 windows 10 PyTorch v1.2 エラー内容 RuntimeError: cuda runtime error(59) : device-side assert triggered 私のソースコードではbackward()の箇所でエラーが発生していました 解…