Yolov5 retraining problem

[nina-vilela]

Hi, re the first warning:

@21010561

This overload of nonzero is deprecated:
nonzero()

This is only a warning, it’s likely not a problem.

And about the RuntimeError, I just made a post that goes through all of the necessary steps for re-training and compiling:

Retrain YOLOv5 on Custom Dataset

This guide will walk through how to retrain YOLOv5 on a custom dataset, in particular a subset of coco-2017 that contains only “person”, “car” and “bicycle” classes. For best practices with different datasets refer to: [image]Tips for Best Training Results Downloading a subset of COCO 2017 Prerequisites: pycotools !pip install git+https://github.com/waleedka/cocoapi.git#egg=pycocotools&subdirectory=PythonAPI fiftyone !pip install fiftyone Download the subset of COCO 2017 with “person”, …

There are steps about creating and editing the dataset that are not relevant to you, but please go through the reamining steps to confirm that everything was done correctly and let us know if you still see the error.