
](
[img]https://i. sstatic.net/H3JmrldO.png[/img]
)
Для отладки рассмотрите возможность передачи CUDA_LAUNCH_BLOCKING=1
Скомпилируйте с TORCH_USE_CUDA_DSA, чтобы включить утверждения на стороне устройства.
Я тренирую YOLOv8 определять атрибуты одежды. Изображения масок и метки ограничивающих рамок передаются в модель для обучения. В эпоху 1 выдает ошибку, показанную на картинках.
Подробнее здесь: https://stackoverflow.com/questions/789 ... rors-might