yolov5 / train.py

Commit History

Add save_dir to plot_lr_scheduler and plot_labels
945307b

astoken commited on

Move results.txt from weights/ to log_dir
3b2b330

astoken commited on

Fix hyp file read in and dict update.
ade023c

astoken commited on

remove old print statements
5f2eeba

astoken commited on

add update default hyp dict with provided yaml
333f678

astoken commited on

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

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