yolov5 / train.py

Commit History

pretrained model loading bug fix (#450)
b569ed6

glenn-jocher commited on

update train.py class count assertion #424
07493a7

glenn-jocher commited on

update train.py class count assertion #424
912c06a

glenn-jocher commited on

update train.py ckpt loading
65857ad

glenn-jocher commited on

comment tb_writer.add_hparams(hyp, {})
03489aa

glenn-jocher commited on

Merge pull request #384 from jancio/master
1e94bcf
unverified

glenn-jocher commited on

Update train.py
120d40c
unverified

glenn-jocher commited on

Fix img_size naming in hyperparameters logging
38acc5f

Janko Ondras commited on

EMA and non_blocking=True
a1c8406

glenn-jocher commited on

Log hyperparameters in tensorboard
e558963

Janko Ondras commited on

multi-gpu ckpt filesize bug fix #253
01a73ec

glenn-jocher commited on

multi-gpu ckpt filesize bug fix #253
a586751

glenn-jocher commited on

update tensorboard metric
5de4e25

glenn-jocher commited on

train.py results.txt to bucket bug fix
98fc483

glenn-jocher commited on

evolution bug fix #346
1c13e67

glenn-jocher commited on

new nc=len(names) check
e16e9e4

glenn-jocher commited on

new nc=len(names) check
cb527d3

glenn-jocher commited on

update log_dir to runs/exp #107
603ea0b

glenn-jocher commited on

disable LR plot to suppress warning message
72d5b58

glenn-jocher commited on

--resume EMA fix #292
24c5a94

glenn-jocher commited on

Move hyp and opt yaml save to top of train()
9d63140
unverified

astoken commited on

hyperparameter printout update
bf6f415

glenn-jocher commited on

Merge branch 'master' into advanced_logging
dc5e183
unverified

glenn-jocher commited on

Update train.py
6b134d9
unverified

glenn-jocher commited on

update train.py and experimental.py
16f6834

glenn-jocher commited on

Add in --resume functionality with option to specify path or to get most recent run
52bac22

astoken commited on

Merge branch 'master' into advanced_logging
f517ba8
unverified

astoken commited on

fix LR bug
956511d
unverified

Laughing commited on

model.yaml nc inherited from dataset.yaml
bb3c346

glenn-jocher commited on

strip_optimizer() bug fix #253
3bdea3f

glenn-jocher commited on

Merge pull request #245 from yxNONG/patch-2
e02a189
unverified

glenn-jocher commited on

Update train.py
597ed4c
unverified

glenn-jocher commited on

Update train.py
1aa2b67
unverified

yxNONG commited on

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