pytorch学习 | 如何统计数据集的均值和标准差?

咱们在使用模型训练以前通常要对数据进行归一化(Normalize),归一化以前须要获得数据集总体的方差和均值,这里提供了一个简单计算数据标准差和均值的接口,方便你们使用。python def get_mean_std(dataset, ratio=0.01): """Get mean and std by sample ratio """ dataloader = torch.ut
相关文章
相关标签/搜索