Написать код Yolov5 (и SSD)

Мне нужно обучить YOLOv5 и SSD (но сначала хотя бы YOLO), а затем описать следующие моменты для моделей: 

Информация по объему данных, использованных для проведения эксперимента - объем, сколько на обучение/валидацию/тест.

Детали, связанные с процессом обучения - сколько эпох, на каком ПК, графики обучения (графики потерь), время обучения.  

Набор данных COCO, но, наверное, нужно рассмотреть какую-то его подвыборку (только не очень маленькую) 

В прикрепленном файле то, что у меня есть, но, вроде, что-то не работает). И, конечно, это довольно срочно

сейчас мне по сути нужно описать эксперимент и указать следующие вещи: нформации по объему данных, использованных для проведения эксперимента - объем, сколько на обучение/валидацию/тест.

и процесс обучения описать: сколько эпох, на каком ПК, графики обучения (графики потерь), время обучения