Skip to content

5 月 2020

[PyTorch] 判斷當前資料所使用的 GPU

Last Updated on 2021-07-05 by Clay

在我使用 PyTorch 訓練模型的時候,經常會發生我使用 GPU_A 去訓練模型、儲存模型,然而在測試模型效果的時候,卻不小心使用到了 GPU_B 來讀取測試資料 (我有多片 GPU 可以使用,還滿奢侈的 XDD),然後再用已經儲存好的模型 (GPU_A 訓練) 來測試 ——

Read More »[PyTorch] 判斷當前資料所使用的 GPU

[已解決][PyTorch] AttributeError: 'tuple' object has no attribute 'size'

Last Updated on 2021-07-05 by Clay

之前,我曾經寫過一篇文章敘述我如何印出我使用 PyTorch 搭建的模型架構,具體連結可以參考文末。但是開心了沒多久,過了一段時間後,當我又要使用這項工具來繪製另一個全新的模型架構準備報告的同時,我卻得到了以下這樣的報錯:

AttributeError: 'tuple' object has no attribute 'size'
Read More »[已解決][PyTorch] AttributeError: 'tuple' object has no attribute 'size'

[已解決][PyTorch] RuntimeError: Expected object of scalar type Float but got scalar type Long for argument

Last Updated on 2021-07-01 by Clay

雖然僅僅只是個人體感,不過我認為在使用 PyTorch 的過程中,最容易遇到的報錯有兩個 —— 一個是模型的 ""Mismatch、另外一個就是我今天所紀錄的:

Read More »[已解決][PyTorch] RuntimeError: Expected object of scalar type Float but got scalar type Long for argument
Exit mobile version