文件名称：pytorch-balanced-batch:pytorch数据集采样器，可始终对平衡批次进行采样
文件大小：14KB
文件格式：ZIP
更新时间：2024-06-16 04:43:02
Python
火炬平衡批次 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个样本
【文件预览】：
pytorch-balanced-batch-master
----LICENSE(34KB)
----README.md(553B)
----sampler.py(2KB)
----.gitignore(17B)
----example.py(604B)