mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
Before, the one_hot could only run in device='cuda'. Now it will run on input device automatically.
Before, the one_hot could only run in device='cuda'. Now it will run on input device automatically.