yolov5 / train.py

Commit History

add logic for resuming and getting hyp for resume run
a448c3b

astoken commited on

add util function to get most recent last.pt file
25e51bc

astoken commited on

add save_dir arg to plot_lr_scheduler, default to current dir.
490f1e7

astoken commited on

change weights dir (wdir) to be unique to each run, under log_dir
4418809

astoken commited on

add save yaml of opt and hyp to tensorboard log_dir in train()
d9f446c

astoken commited on

add parser arg for hyp yaml file
a85e6d0

astoken commited on

check_anchors bug fix
5a50491

glenn-jocher commited on

check_anchors() bug fix #102
05b8ee5

glenn-jocher commited on

check_anchors() bug fix #90
ec81c7b

glenn-jocher commited on

AutoAnchor bug fix #72
8b26e89

glenn-jocher commited on

default check_git_status() to True
915b148

glenn-jocher commited on

FP16 to FP32 ckpt load
14523bb

glenn-jocher commited on

glob search bug fix #77
c5966ab

glenn-jocher commited on

assert best possible recall > 0.9 before training
31f3310

glenn-jocher commited on

Ensures weights/ dir exists
c3d4d32

Lauritzen Kasper Primdal commited on

--img-size stride-multiple verification
099e6f5

glenn-jocher commited on

speed-reproducibility fix #17
22d6088

glenn-jocher commited on

multi-scale fix #16
55ca5c7

glenn-jocher commited on

assert equal model and dataset classes
7c2832c

glenn-jocher commited on