pytorch-image-models icon indicating copy to clipboard operation
pytorch-image-models copied to clipboard

Published 20 hours ago • verified publisher icon huggingface

Fix device problem

Open Leoooo333 opened this issue 2 years ago • 0 comments

Before, the one_hot could only run in device='cuda'. Now it will run on input device automatically.

Leoooo333 avatar Apr 19 '22 03:04 Leoooo333