Easy way to save checkpoints for Colab user #640
Description
🚀 Feature
It will be more convenient for Colab user to save checkpoints in Google Drive than in yolov5/runs.
My idea
Just change in nearly line 458 to 464 (in my current version):
if not opt.evolve:
tb_writer = None
if opt.local_rank in [-1, 0]:
print('Start Tensorboard with "tensorboard --logdir=runs", view at http://localhost:6006/')
# Change the path here
tb_writer = SummaryWriter(log_dir=increment_dir('/content/drive/My Drive/yolov5-checkpoints/exp', opt.name))
train(hyp, opt, device, tb_writer)