上传者: 42097914
|
上传时间: 2021-12-15 21:39:18
|
文件大小: 30KB
|
文件类型: -
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