Error when training yolov8-seg in validators.py

[Landaunn]

Hi!
Describe the bug
I used sparseml-nightly version 1.5.0.20230404 , but I got an error after passing the first epoch of yolov8 segmentation training, which is in the file sparseml/yolov8/validators.py .

Environment

1. OS: Ubuntu 20.04.1
2. Python version: 3.8.10
3. SparseML version or commit hash: sparseml-nightly 1.5.0.20230404
4. ML framework version(s): torch 1.9.1

Errors

[dbogunowicz]

@Landaunn yes, this has come to our attention lately. We know the root cause and will be patching it asap.

[dbogunowicz]

Hey, @Landaunn the fix is in #1522 . We will probably land it today, but if you wish, you can already introduce those changes locally to your repo.