PyTorch中的CutMix：用PyTorch重新实现CutMix
5星
- 浏览量: 0
- 大小:None
- 文件类型：ZIP
简介：
本文介绍了如何使用PyTorch框架重新实现一种名为CutMix的数据增强技术，以提高模型性能。
使用PyTorch实现CutMix的方法如下：
对于模型和其他规范化的代码，请参考相关文档。
训练命令为：
```
python main.py
--print_freq 32 --save_dir .save_model --save_every 10
--lr 0.1 --weight_decay 1e-4 --momentum 0.9
--Epoch 500 --batch_size 128 --test_batch_size 100
--cutout False --n_masks 1 --length 16
--normalize batchnorm
--alpha 1.0 --cutmix_prob 1.0 # For Cutmix
```
实验结果显示，SE + resnet-32 + batchnorm与SE + resnet-32 + batchnorm + cutmix之间存在差异。
全部评论 (0)
还没有任何评论哟~
