yolov5 / train.py

Commit History

small dataset bug fix #140
256a3e8

glenn-jocher commited on

model.names multi-GPU bug fix #94
b50fdf1

glenn-jocher commited on

Fix yaml saving (don't sort keys), reorder --opt keys, bug fix hyp dict accessor
de19165

astoken commited on

Fix bug in --help from percent sign in help string
2d396be

astoken commited on

add stride to datasets.py
b8557f8

glenn-jocher commited on

Turn opt into dictionary before sending it to yaml
611aacf

astoken commited on

Default optimizer SGD
bc4ef48

astoken commited on

Mode all optimizer settings to 'hyp.yaml', integrate proper momentum with Adam optimizer
7abf202

astoken commited on

Fix help message for cfg files
7edbf65

astoken commited on

Remove --resume functionality and related checks/logic.
d64ad0f

astoken commited on

update google_utils import
6c1b87a

glenn-jocher commited on

remove fast, add merge
1f1917e

glenn-jocher commited on

Add plot_results save location to log_dir
e572bb0

astoken commited on

Fix DDP bug in single process multiple device use cases
899f1d4

goodfellowliu commited on

Merge pull request #107 from Lornatang/fix-reference-bugs
cdf1eac
unverified

glenn-jocher commited on

backbone as FP16, save default to FP32
cce95e7

glenn-jocher commited on

save ckpt in FP16 #119
d9b64c2

glenn-jocher commited on

AutoAnchor bug fix # 117
9fdb0fb

glenn-jocher commited on

Syntax fixes
c8152c8

astoken commited on

Add save_dir arg to test.test, use arg as location for saving batch jpgs
9b7386f

astoken commited on

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

dist.destroy_process_group() bug fix
afe1df3

glenn-jocher 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

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