python - Google Colab Notebook completely freezes when training a YOLO model

Question

I am following a tutorial to train custom object detection model using YOLO. This is the tutorial and also where I got the Notebook

Everything works fine until the training bit which is the last cell. It complete a few iterations then it freezes and the page stops responding. I’m in a little hurry so any way to fix this is appreciated. TIA

Answer 1

from IPython.utils import io
with io.capture_output() as captured:
   !./darknet detector train data/obj.data cfg/yolov3_training.cfg darknet53.conv.74 - 
   dont_show

I had same issue, this one worked for me!