Hugging Face's logo

Spaces:
xiang-wuu
/
yolov5
Runtime error

App Files Community
yolov5 / train.py

Commit History

--resume bug fix #252
86784cf

glenn-jocher commited on

--resume bug fix #187
ad4c22c

glenn-jocher commited on

Unify the check point of single and multi GPU
cdb9bde
unverified

yxNONG commited on

update train.py incompatible model message fix #222
b203c9b

glenn-jocher commited on

update mosaic border
37e13f8

glenn-jocher commited on

Merge branch 'master' into advanced_logging
e18e681
unverified

astoken commited on

refactor dataloader
22fb2b0

glenn-jocher commited on

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