pytorch tensor int型除法出現(xiàn)的問題
昨天晚上跑起來(lái)一個(gè)classification實(shí)驗(yàn),今天發(fā)現(xiàn)訓(xùn)練loss在降,然而accuracy永遠(yuǎn)是0 。。。直覺告訴我evaluation有問題
然后發(fā)現(xiàn)自己寫了個(gè)很愚蠢的bug
accuracy對(duì)應(yīng)的tensor出來(lái)是int型的,我用到了一個(gè)除法取平均。而pytorch里無(wú)論用 / or // 結(jié)果都是取整,,accuracy一直就沒有。。
所以轉(zhuǎn)換成float就沒問題了,,低級(jí)bug,下不為例
補(bǔ)充:pytorch tensor division/除法
除法的時(shí)候要注意數(shù)據(jù)類型的問題
a = torch.div(torch.tensor([160, 110]), 0.137)
jupyter notebook就會(huì)奔潰, 因?yàn)閠orch.tensor([160, 110])是torch.int64數(shù)據(jù)類型, 而0.137(或者說(shuō)torch.tensor(0.137))是torch.float32類型.
如果改成下面這樣, 就會(huì)順利運(yùn)行
a = torch.div(torch.tensor([160, 110]).float(), 0.137)
以上為個(gè)人經(jīng)驗(yàn),希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。如有錯(cuò)誤或未考慮完全的地方,望不吝賜教。
相關(guān)文章
Pygame框架實(shí)現(xiàn)飛機(jī)大戰(zhàn)
這篇文章主要為大家詳細(xì)介紹了Pygame框架實(shí)現(xiàn)飛機(jī)大戰(zhàn),文中示例代碼介紹的非常詳細(xì),具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2020-08-08Jinja2實(shí)現(xiàn)模板渲染與訪問對(duì)象屬性流程詳解
要了解jinja2,那么需要先理解模板的概念。模板在Python的web開發(fā)中廣泛使用,它能夠有效的將業(yè)務(wù)邏輯和頁(yè)面邏輯分開,使代碼可讀性增強(qiáng),并且更加容易理解和維護(hù)。模板簡(jiǎn)單來(lái)說(shuō)就是一個(gè)其中包含占位變量表示動(dòng)態(tài)部分的文,模板文件在經(jīng)過(guò)動(dòng)態(tài)賦值后,返回給用戶2023-03-03數(shù)據(jù)清洗--DataFrame中的空值處理方法
今天小編就為大家分享一篇數(shù)據(jù)清洗--DataFrame中的空值處理方法,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-07-07python高級(jí)搜索實(shí)現(xiàn)高效搜索GitHub資源
這篇文章主要為大家介紹了python高級(jí)搜索來(lái)高效搜索GitHub,從而高效獲取所需資源,有需要的朋友可以借鑒參考下,希望能夠有所幫助2021-11-11Django URL和View的關(guān)系說(shuō)明
這篇文章主要介紹了Django URL和View的關(guān)系說(shuō)明,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2021-03-03淺析python打包工具distutils、setuptools
python包在開發(fā)中十分常見,一般的使用套路是所有的功能做一個(gè)python模塊包,打包模塊,然后發(fā)布,安裝使用。這篇文章給大家介紹了python打包工具distutils、setuptools的相關(guān)知識(shí),感興趣的朋友一起看看吧2018-04-04Python使用pylab庫(kù)實(shí)現(xiàn)繪制直方圖功能示例
這篇文章主要介紹了Python使用pylab庫(kù)實(shí)現(xiàn)繪制直方圖功能,結(jié)合實(shí)例形式分析了Python數(shù)據(jù)讀取、遍歷以及基于pylab庫(kù)繪制直方圖的相關(guān)操作技巧,需要的朋友可以參考下2018-06-06