火炬平衡批次
pytorch数据集sampler用于始终采样平衡批次。
确保使用的batch_size是类数的整数倍。
例如,如果您的train_dataset有10个类,并且您将batch_size=30与BalancedBatchSampler
train_loader = torch . utils . data . DataLoader ( train_dataset , sampler = BalancedBatchSampler ( train_dataset ), batch_size = 30 )
您将获得一个train_loader ,其中10个类中的每个类的每个元素都有3个样本
2022-06-13 15:37:23
14KB
Python
1