pytorch tensor int型除法出現(xiàn)的問題
更新時間：2021年04月08日 14:17:53 作者：judgechen1997
這篇文章主要介紹了pytorch tensor int型除法出現(xiàn)的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
昨天晚上跑起來一個classification實驗，今天發(fā)現(xiàn)訓(xùn)練loss在降，然而accuracy永遠是0 。。。直覺告訴我evaluation有問題
然后發(fā)現(xiàn)自己寫了個很愚蠢的bug
accuracy對應(yīng)的tensor出來是int型的，我用到了一個除法取平均。而pytorch里無論用 / or // 結(jié)果都是取整，，accuracy一直就沒有。。
所以轉(zhuǎn)換成float就沒問題了，，低級bug，下不為例
補充：pytorch tensor division/除法
除法的時候要注意數(shù)據(jù)類型的問題
a = torch.div(torch.tensor([160, 110]), 0.137)
jupyter notebook就會奔潰, 因為torch.tensor([160, 110])是torch.int64數(shù)據(jù)類型, 而0.137(或者說torch.tensor(0.137))是torch.float32類型.
如果改成下面這樣, 就會順利運行
a = torch.div(torch.tensor([160, 110]).float(), 0.137)
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。
相關(guān)文章
Django配置celery（非djcelery）執(zhí)行異步任務(wù)和定時任務(wù)
這篇文章主要介紹了Django配置celery（非djcelery）執(zhí)行異步任務(wù)和定時任務(wù)，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-07-07
這篇文章主要介紹了如何利用python執(zhí)行txt文件中的代碼，python這么強大的語言當然可以做大，只需使用內(nèi)置的exex()函數(shù)，進入主題前我們先來看看什么是exec()函數(shù)，需要的小伙伴可以參考一下2022-03-03
這篇文章主要介紹了python實現(xiàn)的希爾排序算法,實例分析了基于Python實現(xiàn)希爾排序的相關(guān)技巧,需要的朋友可以參考下2015-07-07
Django數(shù)據(jù)映射(一對一,一對多,多對多)
本文主要介紹了Django數(shù)據(jù)映射(一對一,一對多,多對多)，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2022-08-08
本文主要介紹了Pycharm搭建一個Django項目的方法步驟，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2023-02-02
時間戳是一個表示日期和時間的數(shù)值,通常以秒為單位,在Python中,獲取時間戳是常見的任務(wù),用于記錄事件、計時操作、以及在各種應(yīng)用中跟蹤時間,本文將介紹多種獲取時間戳的方法,包括標準庫和第三方庫的方式,并提供示例代碼以幫助你更好地理解2023-11-11
Python中列表乘法和列表推導(dǎo)式的區(qū)別舉例詳解
在Python中列表是一種非常靈活和強大的數(shù)據(jù)結(jié)構(gòu),支持多種運算和操作,這篇文章主要介紹了Python中列表乘法和列表推導(dǎo)式區(qū)別的相關(guān)資料,文中通過代碼就介紹的非常詳細,需要的朋友可以參考下2025-04-04
Django權(quán)限系統(tǒng)auth模塊用法解讀
這篇文章主要介紹了Django權(quán)限系統(tǒng)auth模塊用法解讀，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2023-05-05
基于python requests selenium爬取excel vba過程解析
這篇文章主要介紹了基于python requests selenium爬取excel vba過程解析,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-08-08
這篇文章主要介紹了python正則表達式的懶惰匹配和貪婪匹配說明，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-07-07