pytorch學習 | 如何統計數據集的均值和標準差?

咱們在使用模型訓練以前通常要對數據進行歸一化(Normalize),歸一化以前須要獲得數據集總體的方差和均值,這裏提供了一個簡單計算數據標準差和均值的接口,方便你們使用。python def get_mean_std(dataset, ratio=0.01): """Get mean and std by sample ratio """ dataloader = torch.ut
相關文章
相關標籤/搜索