Pytorch1 0 0 有哪些坑?

時間 2021-05-05 22:58:14

1樓:打柴桑

# 先轉FloatTensor才能.cuda()input_tensor = input_tensor.type(torch.FloatTensor).cuda()

# 先.cuda()然後 FloatTensorinput_tensor = input_tensor.cuda().type(torch.FloatTensor)

報錯「RuntimeError: Expected object of backend CPU but got backend CUDA「

2樓:可達鴨

torch.sparse.mm 在 torch==1.0.1 版本下有時候居然會出現多次運算結果不一致且為 NaN 的情況。如下圖:

torch.sum(torch.isnan(torch.sparse.mm(adj, support)))

Out[22]: tensor(0)

torch.sum(torch.isnan(torch.sparse.mm(adj, support)))

Out[23]: tensor(2)

torch.sum(torch.isnan(torch.sparse.mm(adj, support)))

Out[24]: tensor(0)

我的 adj 引數只是乙個全為 0 的稀疏矩陣。。這個破問題讓我 debug 了整個模型幾個小時。

3樓:向日葵

torchvision.transforms的Resize的size引數在torch0.3.

1和0.4.0之間的height和width居然是相反設定,害我debug了好久。。。

4樓:王晉東不在家

乙個很不爽的設定:

不同的資料型別的tensor不能拼接。

比如我有一列a是label(整數),乙個b矩陣是feature matrix(float),切分資料時想要合併起來,很自然想到torch.cat((a,b), dim=1),然後PyTorch說資料型別不同不能合併。。。

5樓:Big Fish

6樓:Littlemite

用jit.trace的時候train和eval模式要分別trace

ScriptModule的module 成員不會自動變成ScriptModule. 所以如果要把已有模型變成完全的ScriptModule還是有點麻煩的。

此外,bert暫時未能trace成功…

7樓:周軍

目前crnn直接改一下ctcloss 並且加上在輸出上log_softmax會遇到一定epoch後nan的問題,還在排查中,不知道算不算坑

開車行駛100邁過坑 對車都有哪些影響?

講一下自己的事吧,在高中的時候,自己有一次騎自行車,沒注意到有乙個坑,當時速度可能最快四十,人都直接飛出去了。還有一次開車的時候,遇到坑躲避不及,當時好像是七十的速度,人都直接彈起來了。對車傷害具體是在什麼地方我不知道,但是對駕駛員還是有點心跳的感覺哦。傷輪胎 輪轂。懸架視情況。曾經騎電單車有過類似...

推薦系統有哪些坑?

神策資料 前面有很多優質回答,我就補充2個前面沒有的實踐經驗以及解決方案 某一次使用推薦系統進行小的流量測試上線之後,技術同事發現效果不如預期,便進行了詳細的資料分析,發現了兩個方面的資料異常。第一,命中行為模型的使用者較少。通常情況下,只要不是乙個新使用者,理論上來說,都應該能夠命中該行為模型。當...

有哪些好用鋼筆推薦,100以下,100到300,300到1000的,每個檔位能否都推薦一些?

葉七 100,百樂78g,福林812 注意這是一款打磨尖 300 英雄100 京東活動價150 末匠f9 1000 寫樂大型21k mf 其實m更好,但是有些人人被m嚇到,實際不粗 百樂742 Schickelgruber 100以下基本就百樂了吧。100到300的話佩爾娜?還是百樂 Lamy或者施...