Fix cutmix augmentation when minmax is specified

facebookresearch / ClassyVision

An end-to-end PyTorch framework for image and video classification

https://classyvision.ai

MIT License

1.59k stars 278 forks source link

Closed edpizzi closed 3 years ago

edpizzi commented 3 years ago

Summary: The issue is that the code provides max as a tensor to torch.randint, which expects only scalars, unlike the numpy equivalent.

Differential Revision: D30761197

facebook-github-bot commented 3 years ago

This pull request was exported from Phabricator. Differential Revision: D30761197