yolov5 / train.py

Commit History

Evolve in CSV format (#4307)
e78aeac
unverified

glenn-jocher commited on

Feature `python train.py --cache disk` (#4049)
2d99063
unverified

junji hashimoto glenn-jocher commited on

Add `train.py` and `val.py` callbacks (#4220)
b74929c
unverified

Kalen Michael glenn-jocher commited on

Add `python train.py --freeze N` argument (#4238)
bceb57b
unverified

IneovaAI glenn-jocher commited on

Fix weight decay comment (#4228)
8d3c3ef
unverified

glenn-jocher commited on

Explicit `requirements.txt` location (#4225)
c2c958c
unverified

glenn-jocher commited on

PyCharm reformat (#4209)
b60b62e
unverified

glenn-jocher commited on

W&B: Restructure code to support the new dataset_check() feature (#4197)
e88e8f7
unverified

Ayush Chaurasia glenn-jocher commited on

Train from `--data path/to/dataset.zip` feature (#4185)
5d66e48
unverified

glenn-jocher commited on

Update script headers (#4163)
0ad6301
unverified

glenn-jocher commited on

New CSV Logger (#4148)
96e36a7
unverified

glenn-jocher commited on

Refactor train.py and val.py `loggers` (#4137)
efe60b5
unverified

glenn-jocher commited on

Update train.py (#4136)
63dd65e
unverified

glenn-jocher commited on

Add train.py ``--img-size` floor (#4099)
2c073cd
unverified

glenn-jocher commited on

`val.py` refactor (#4053)
f7d8562
unverified

glenn-jocher Ayush Chaurasia commited on

Add `--sync-bn` known issue (#4032)
951922c
unverified

glenn-jocher commited on

Rename `test.py` to `val.py` (#4000)
720aaa6
unverified

glenn-jocher commited on

Fix inconsistent NMS IoU value for COCO (#3934)
e7888af
unverified

ekurtic commited on

Evolution commented `hyp['anchors']` fix (#3887)
8930e22
unverified

glenn-jocher commited on

`--evolve 300` generations CLI argument (#3863)
d3e9d69
unverified

san-soucie sansoucie glenn-jocher commited on

Copy-Paste augmentation for YOLOv5 (#3845)
c6c88dc
unverified

glenn-jocher commited on

Fix warmup `accumulate` (#3722)
3974d72
unverified

yellowdolphin glenn-jocher commited on

Update seeds for single-GPU reproducibility (#3789)
92d49fd
unverified

glenn-jocher commited on

fix/incorrect_fitness_import (#3770)
09246a5
unverified

Piotr Skalski commited on

Remove DDP MultiHeadAttention fix (#3768)
f2d97eb
unverified

glenn-jocher commited on

Add optional dataset.yaml `path` attribute (#3753)
f79d747
unverified

glenn-jocher commited on

Force non-zero hyp evolution weights `w` (#3748)
ae4261c
unverified

glenn-jocher commited on

Create `data/hyps` directory (#3747)
fdc2239
unverified

glenn-jocher commited on

Update 4 main ops for paths and .run() (#3715)
1f69d12
unverified

glenn-jocher commited on

[x]W&B: Don't resume transfer learning runs (#3604)
75c0ff4
unverified

Ayush Chaurasia commited on

Update DDP backend `if dist.is_nccl_available()` (#3705)
e8810a5
unverified

glenn-jocher commited on

Add `train.run()` method (#3700)
fbf41e0
unverified

glenn-jocher commited on

Add torch DP warning (#3698)
c1af67d
unverified

glenn-jocher commited on

Eliminate `total_batch_size` variable (#3697)
b3e2f4e
unverified

glenn-jocher commited on

Update DDP for `torch.distributed.run` with `gloo` backend (#3680)
fad27c0
unverified

glenn-jocher commited on

Slightly modify CLI execution (#3687)
bfb2276
unverified

lb-desupervised Lewis Belcher commited on

Update `WORLD_SIZE` and `RANK` retrieval (#3670)
2296f15
unverified

glenn-jocher commited on

Update TensorBoard (#3669)
045d5d8
unverified

glenn-jocher commited on

Update `train(hyp, *args)` to accept `hyp` file or dict (#3668)
fa201f9
unverified

glenn-jocher commited on

Update train.py (#3667)
6d6e2ca
unverified

glenn-jocher commited on

Fix incorrect end epoch comment (#3612)
4c5d9bf
unverified

Wei Quan commited on

train.py GPU memory fix (#3590)
4984cf5
unverified

glenn-jocher commited on

Refactoring cleanup (#3565)
4695ca8
unverified

glenn-jocher commited on

Update test.py profiling (#3555)
5948f20
unverified

glenn-jocher commited on

Remove `is_coco` argument from `test()` (#3553)
63157d2
unverified

glenn-jocher commited on

Remove `opt` from `create_dataloader()`` (#3552)
958ab92
unverified

glenn-jocher commited on

On-demand `pycocotools` pip install (#3547)
ef0b5c9
unverified

glenn-jocher commited on

Merge `develop` branch into `master` (#3518)
f3c3d2c
unverified

glenn-jocher Piotr Skalski SkalskiP Peretz Cohen tudoulei chocosaj BuildTools Yonghye Kwon Sam_S Samridha Shrestha edificewang commited on

TensorBoard DP/DDP graph fix (#3325)
aad99b6
unverified

glenn-jocher commited on

Improves docs and handling of entities and resuming by WandbLogger (#3264)
19100ba
unverified

Charles Frye commited on