yolov5 / train.py

Commit History

Fix missing model.stride in DP and DDP mode (#683)
3d8ed0a
unverified

Nanobit commited on

Single-source training update (#680)
a0ac5ad

glenn-jocher commited on

Single-source training (#680)
3c6e2f7
unverified

glenn-jocher commited on

Fix unrecognized local rank argument (#676)
d7cfbc4
unverified

Nanobit commited on

train.py --logdir argparser addition (#660)
9368453
unverified

glenn-jocher commited on

Add Multi-Node support for DDP Training (#504)
886b984
unverified

Nanobit commited on

Fix missing imports (#627)
728efa6
unverified

lorenzomammana glenn-jocher commited on

import random bug fix (#614)
eb99dff

glenn-jocher commited on

Explicit Imports (#498)
d5b6416
unverified

Jirka Borovec commited on

hyperparameter evolution update (#566)
f1096b2

glenn-jocher commited on

hyperparameter evolution bug fix (#566)
c1a2a7a

glenn-jocher commited on

hyperparameter evolution bug fix (#566)
8074745

glenn-jocher commited on

hyperparameter evolution bug fix (#566)
e32abb5

glenn-jocher commited on

hyperparameter evolution bug fix (#566)
8056fe2

glenn-jocher commited on

hyperparameter expansion to flips, perspective, mixup
127cbeb

glenn-jocher commited on

replace random_affine() with random_perspective()
bcd452c

glenn-jocher commited on

PyTorch 1.6.0 update with native AMP (#573)
c020875
unverified

goodfellowliu commited on

LR --resume repeat bug fix (#565)
4e2b9ec
unverified

Laughing commited on

Fix bug #541 #542 (#545)
a209a32
unverified

AlexWang1900 glenn-jocher commited on

v2.0 Release (#491)
9da56b6
unverified

glenn-jocher commited on

Update train.py (#462)
5e970d4
unverified

glenn-jocher commited on

update train.py gsutil bucket fix (#463)
3edc38f

glenn-jocher commited on

update train.py gsutil bucket fix (#463)
7765557

glenn-jocher commited on

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