App下載

pytorch為什么要使用float64訓練?怎么使用?

你是我的所有夢 2021-07-15 11:54:42 瀏覽數 (5966)
反饋

在神經網絡的推理過程中,我們默認是將pytorch的tensor的精度指定為單精度的float32,但是在有些時候不管是float16還是float32,在訓練模型的時候都會導致精度丟失,從而引發(fā)訓練效果的損失。那么pytorch怎么使用float64訓練呢?接下來這篇文章告訴你。

首先我們要知道,pytorch默認使用單精度float32訓練模型,

原因在于:

使用float16訓練模型,模型效果會有損失,而使用double(float64)會有2倍的內存壓力,且不會帶來太多的精度提升。

本人,最近遇到需要使用double數據類型訓練模型的情況,具體實現需要把模型的權重參數數據類型和輸入數據類型全部設置為torch.float64即可。

可使用torch的一個函數,輕松地把模型參數轉化為float64

torch.set_default_dtype(torch.float64)

輸入類型可使用

tensor.type(torch.float64)

補充:float32和float64的本質區(qū)別

首先我們需要知道何為bits和bytes?

bits:名為位數bytes:為字節(jié)簡單的數就是MB和G的關系!

那么8bits=1bytes,下面是各個單位的相互轉化!

存儲單位轉化

那么float32和float64有什么區(qū)別呢?

數位的區(qū)別一個在內存中占分別32和64個bits,也就是4bytes或8bytes數位越高浮點數的精度越高它會影響深度學習計算效率?

float64占用的內存是float32的兩倍,是float16的4倍;

比如對于CIFAR10數據集,如果采用float64來表示,需要60000*32*32*3*8/1024**3=1.4G,光把數據集調入內存就需要1.4G;

如果采用float32,只需要0.7G,如果采用float16,只需要0.35G左右;

占用內存的多少,會對系統運行效率有嚴重影響;(因此數據集文件都是采用uint8來存在數據,保持文件最小)

小結

在了解了float32和float64有什么區(qū)別之后,相信各位讀者心中已經知道應該如何選擇訓練模型時要使用哪種精度了。以上為個人經驗,希望能給大家一個參考,也希望大家多多支持W3Cschool。



0 人點贊