yolov5 / train.py

Commit History

Refactor collections and fstrings (#7821)
f000714
unverified

glenn-jocher commited on

Add `@threaded` decorator (#7813)
4a295b1
unverified

glenn-jocher pre-commit-ci[bot] commited on

Rename `utils/datasets.py` > `utils/dataloaders.py` (#7799)
9d8ed37
unverified

glenn-jocher commited on

Implement DDP `static_graph=True` (#6940)
d95a728
unverified

glenn-jocher pre-commit-ci[bot] commited on

Update `on_train_end` callback (#7716)
aa7a0e9
unverified

glenn-jocher commited on

Remove `tqdm.auto` (#7599)
b53917d
unverified

glenn-jocher commited on

PyTorch Hub `_verbose=False` fix2 (#7550)
7043872
unverified

glenn-jocher pre-commit-ci[bot] commited on

Add `--noplots` flag to suppress figures and images logging (#7534)
4cac1cf
unverified

Ayush Chaurasia pre-commit-ci[bot] glenn-jocher commited on

Fix val `plots=plots` (#7524)
23718df
unverified

glenn-jocher commited on

Improved non-latin `Annotator()` plotting (#7488)
c9042dc
unverified

glenn-jocher pre-commit-ci[bot] commited on

Dynamic normalization layer selection (#7392)
4bb7eb8
unverified

glenn-jocher commited on

Add support for different normalization layers (#7377)
fa569cd
unverified

Vardan Agarwal pre-commit-ci[bot] glenn-jocher commited on

Update optimizer param group strategy (#7376)
bd2dda8
unverified

glenn-jocher commited on

Print dataset scan only `if RANK in (-1, 0)` (#7337)
5f941a8
unverified

glenn-jocher pre-commit-ci[bot] commited on

Add callbacks (#7315)
245d645
unverified

glenn-jocher commited on

Use `tqdm.auto` (#7311)
f735458
unverified

glenn-jocher commited on

CLI `fire` prep updates (#7229)
2c32218
unverified

glenn-jocher commited on

precommit: yapf (#5494)
c3d5ac1
unverified

Jirka Borovec pre-commit-ci[bot] glenn-jocher commited on

Add train.py `--name cfg` option (#7202)
df9008e
unverified

glenn-jocher commited on

Update min warmup iterations from 1k to 100 (#6768)
47f265d
unverified

glenn-jocher commited on

Updated VOC hyperparameters (#6732)
2692e67
unverified

glenn-jocher commited on

Switch default LR scheduler from cos to linear (#6729)
a936f5f
unverified

glenn-jocher commited on

Fix `--evolve --bucket gs://...` (#6698)
3670d91
unverified

glenn-jocher commited on

Add `--cache val` (#6663)
7b80545
unverified

glenn-jocher commited on

Validate with 2x `--workers` (#6658)
6769021
unverified

glenn-jocher commited on

Update train.py
a5c9057
unverified

glenn-jocher commited on

return `opt` from `train.run()` (#6581)
9cf80b7
unverified

Christoph Fürbacher commited on

Allow custom` --evolve` project names (#6567)
f40854b
unverified

Matthias Vadcard glenn-jocher commited on

Improved AutoBatch DDP error message (#6568)
006eb40
unverified

glenn-jocher commited on

Load checkpoint on CPU instead of on GPU (#6516)
aff0281
unverified

bilzard glenn-jocher commited on

W&B: Remember batchsize on resuming (#6512)
b73c62e
unverified

Ayush Chaurasia glenn-jocher commited on

Add `stop_training=False` flag to callbacks (#6365)
8efe977
unverified

Matthias glenn-jocher commited on

Flush callbacks when on `--evolve` (#6374)
9bcc32a
unverified

Ayush Chaurasia glenn-jocher commited on

Fix `train.py` parameter groups desc error (#6318)
db1f83b
unverified

Otfot glenn-jocher commited on

Update P2-P7 `models/hub` variants (#6230)
33a67b4
unverified

glenn-jocher commited on

Update export format docstrings (#6151)
ec4b6dd
unverified

glenn-jocher pre-commit-ci[bot] commited on

Enable AdamW optimizer (#6152)
e1dc894
unverified

bilzard commited on

W&B: track batch size after autobatch (#6039)
db6ec66
unverified

Ayush Chaurasia glenn-jocher commited on

`--freeze` fix (#6044)
dc54ed5
unverified

glenn-jocher commited on

Multi-layer capable `--freeze` argument (#6019)
361705d
unverified

Felix You glenn-jocher commited on

Remove NCOLS from tqdm (#5804)
7c6bae0
unverified

glenn-jocher pre-commit-ci[bot] commited on

W&B: refactor W&B tables (#5737)
f2ca30a
unverified

Ayush Chaurasia pre-commit-ci[bot] glenn-jocher commited on

Do not save hyp.yaml and opt.yaml on evolve (#5775)
4ca4aec
unverified

glenn-jocher commited on

Update train.py comment to 'Model attributes' (#5670)
47fac9f
unverified

glenn-jocher commited on

Default DataLoader `shuffle=True` for training (#5623)
09d1703
unverified

Werner Duvaud glenn-jocher pre-commit-ci[bot] commited on

DDP `WORLD_SIZE`-safe dataloader workers (#5631)
7473f0f
unverified

glenn-jocher commited on

Update train, val `tqdm` to fixed width (#5367)
61c5019
unverified

glenn-jocher pre-commit-ci[bot] commited on

`LOGGER` consolidation (#5569)
79bca2b
unverified

glenn-jocher pre-commit-ci[bot] commited on

`intersect_dicts()` in hubconf.py fix (#5542)
e189fa1
unverified

glenn-jocher commited on

Common `is_coco` logic betwen train.py and val.py (#5521)
bcc085d
unverified

glenn-jocher commited on