* Fix torch GPU error

* Update torch_utils.py

single-line device =
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>