我正在嘗試使用 pytorch 訓練 DNN 模型,并且我想使用 GPU 來訓練我的模型。我能夠成功地使用 將我的模型復制到 GPU model.to(device),其中device = cuda:0.
但是,將輸入復制到 GPU 的標準方法,(
請讓我知道我哪里出錯了。謝謝!
uj5u.com熱心網友回復:
X.to(device)什么也沒做。將其更改為:
x=x.to(device)
當然,這應該對您想要在 GPU 上的任何引數\變數進行
轉載請註明出處,本文鏈接:https://www.uj5u.com/qukuanlian/364542.html
上一篇:標記化和分段之間的區別
