找不到mixup_batch函数

[Chy-ang]

ghostnetv2_pytorch/train.py中的from timm.data import Dataset, create_loader, resolve_data_config, FastCollateMixup, mixup_batch, AugMixDataset这里的mixup_batch找不到这个函数，请问是什么问题

[thkey]

同求，我也找不到

[xie103]

解决了吗，我也有这个问题

[iamhankai]

可以用这个：https://github.com/alvinwan/nbdt-pytorch-image-models/blob/fffcb25630458c0c82cba546b5d174f2c98ad25c/timm/data/mixup.py#L18

[xie103]

可以用这个：https://github.com/alvinwan/nbdt-pytorch-image-models/blob/fffcb25630458c0c82cba546b5d174f2c98ad25c/timm/data/mixup.py#L18

是替换mixup.py这一个文件吗？还是怎样

[Neo-code003]

可以用这个：https://github.com/alvinwan/nbdt-pytorch-image-models/blob/fffcb25630458c0c82cba546b5d174f2c98ad25c/timm/data/mixup.py#L18

是替换mixup.py这一个文件吗？还是怎样

把里面的def mixup_batch 和FastCollateMixup里的 init 直接复制过去 再把FastCollateMixup里的__call__替换掉

然后再把data/init.py里的Mixup改为mixup_batch就行了