clovaai / CutMix-PyTorch

Official Pytorch implementation of CutMix regularizer
MIT License
1.22k stars 159 forks source link

some problems about the parameters #23

Closed ReverseSystem001 closed 4 years ago

ReverseSystem001 commented 4 years ago

thanks for your amazing paper . follow the paper. I set the beta = 1. what number the cutmix_prob should i set?

hellbell commented 4 years ago

@ReverseSystem001 cutmix_prob is set to 1.0 in the default setting and it usually works well. Empirically, we set it to 0.5 or lower when the network is lightweight, such as mobilenet, but it depends on what dataset is used.