pytorch--CutMix(pytorch)
用火炬实施
有关模型和规范化的其他代码,请参见
为了训练
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之间的
2021-10-27 20:35:03
3.57MB
Python
1