pytorch-randaugment
RandAugment的非官方PyTorch重新实现。 大部分代码来自 。
介绍
可以使用RandAugment对感兴趣的数据集训练模型,而无需单独的代理任务。 通过仅调整两个超参数(N,M),您可以实现具有竞争优势的AutoAugments性能。
安装
$ pip install git+https://github.com/ildoonet/pytorch-randaugment
用法
from torchvision . transforms import transforms
from RandAugment import RandAugment
transform_train = transforms . Compose ([
transforms . RandomCrop ( 32 , padding = 4 ),
t
1