我们在使用模型训练之前一般要对数据进行归一化(Normalize),归一化之前需要得到数据集整体的方差和均值,这里提供了一个简单计算数据标准差和均值的接口,方便大家使用。
def get_mean_std(dataset, ratio=0.01):
"""Get mean and std by sample ratio
"""
dataloader = torch.utils.data.DataLoader(dataset, batch_size=int
我们在使用模型训练之前一般要对数据进行归一化(Normalize),归一化之前需要得到数据集整体的方差和均值,这里提供了一个简单计算数据标准差和均值的接口,方便大家使用。
def get_mean_std(dataset, ratio=0.01):
"""Get mean and std by sample ratio
"""
dataloader = torch.utils.data.DataLoader(dataset, batch_size=int