yolov5 / train.py

Commit History

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

Fix TypeError: 'PosixPath' object is not iterable (#3285)
dd7f0b7
unverified

glenn-jocher commited on

Assert `--image-weights` not combined with DDP (#3275)
10d56d7
unverified

glenn-jocher commited on

TensorBoard add_graph() fix (#3236)
b7cd1f5
unverified

glenn-jocher commited on

Update train.py (#3099)
60fe544
unverified

glenn-jocher commited on

Update check_requirements() exclude list (#2974)
a833ee2
unverified

glenn-jocher commited on

Implement yaml.safe_load() (#2876)
f7bc685
unverified

glenn-jocher commited on

Detection cropping+saving feature addition for detect.py and PyTorch Hub (#2827)
c949fc8
unverified

burhan glenn-jocher commited on

Tensorboard model visualization bug fix (#2758)
6dd1083
unverified

glenn-jocher commited on

Add TransformerLayer, TransformerBlock, C3TR modules (#2333)
1148e2e
unverified

Ding Yiwei DingYiwei glenn-jocher commited on

Add --label-smoothing eps argument to train.py (default 0.0) (#2344)
9c803f2
unverified

Phat Tran phattran glenn-jocher commited on

W&B resume ddp from run link fix (#2579)
518c095
unverified

Ayush Chaurasia commited on

Fix: evolve with wandb (#2634)
dc51e80
unverified

Ayush Chaurasia commited on

W&B DDP fix 2 (#2587)
9f98201
unverified

glenn-jocher commited on

Update tensorboard>=2.4.1 (#2576)
e5b0200
unverified

glenn-jocher commited on

W&B DDP fix (#2574)
1bf9365
unverified

Ayush Chaurasia commited on

Improved W&B integration (#2125)
e8fc97a
unverified

Ayush Chaurasia glenn-jocher commited on

labels.jpg class names (#2454)
08d4918
unverified

glenn-jocher commited on

Integer printout (#2450)
f01f322
unverified

glenn-jocher commited on

DDP after autoanchor reorder (#2421)
886f1c0
unverified

glenn-jocher commited on

Resume with custom anchors fix (#2361)
e931b9d
unverified

glenn-jocher commited on

Anchor override (#2350)
a3ecf0f
unverified

glenn-jocher commited on

EMA bug fix 2 (#2330)
fab5085
unverified

glenn-jocher commited on

final_epoch EMA bug fix (#2317)
dfeec19
unverified

glenn-jocher commited on

W&B entity support (#2298)
efa4946
unverified

Jan Hajek Jan Hajek glenn-jocher commited on