yolov5 / train.py

Commit History

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

precommit: isort (#5493)
0155548
unverified

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

Fix float zeros format (#5491)
5866646
unverified

glenn-jocher commited on

Update `get_loggers()` (#4854)
7b1f7ae
unverified

glenn-jocher pre-commit commited on

DDP `nl` fix (#5332)
a4fece8
unverified

glenn-jocher commited on

Add `autobatch` feature for best `batch-size` estimation (#5092)
ca19df5
unverified

glenn-jocher commited on

Add `on_fit_epoch_end` callback (#5232)
752d97d
unverified

glenn-jocher commited on

Fix `ROOT` as relative path (#5129)
153873e
unverified

Malte Lorbach glenn-jocher commited on

Pass `LOCAL_RANK` to `torch_distributed_zero_first()` (#5114)
4a6dfff
unverified

Nan qiningonline commited on

Fix `yaml.safe_load()` ignore emoji errors (#5060)
070af88
unverified

glenn-jocher commited on

Implement `--save-period` locally (#5047)
5afc9c2
unverified

glenn-jocher commited on

Fix pylint: do not use bare 'except' (#5025)
1922dde
unverified

Zhiqiang Wang commited on

Compute loss on final val (#5017)
89a6e9f
unverified

glenn-jocher commited on

Update train.py (#5014)
452e847
unverified

glenn-jocher commited on

Allow YOLOv5 execution from arbitrary `cwd` (#4954)
4e04cb0
unverified

glenn-jocher commited on

Validate `best.pt` on train end (#4889)
d856c48
unverified

glenn-jocher commited on

Fix DDP destruction `LOGGER.info()` (#4863)
2b3109e
unverified

glenn-jocher commited on

Simplify `check_requirements()` usage (#4855)
4c839ee
unverified

glenn-jocher commited on

Update `sys.path.append(str(ROOT))` (#4852)
4d1a2ac
unverified

glenn-jocher commited on