yolov5 / utils /torch_utils.py

Commit History

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

glenn-jocher Ayush Chaurasia commited on

Hub models `map_location=device` (#3894)
6a3ee7c
unverified

glenn-jocher commited on

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

glenn-jocher commited on

Multi-GPU default to single device 0 (#3554)
8b5086c
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

check `batch_size % utilized_device_count` (#3276)
f340235
unverified

glenn-jocher commited on

Speed profiling improvements (#2648)
866bc7d
unverified

glenn-jocher commited on

Improve git_describe() fix 1 (#2635)
2e95cf3
unverified

glenn-jocher commited on

Improve git_describe() (#2633)
6e8c5b7
unverified

glenn-jocher commited on

Create date_modified() (#2616)
0ff5aec
unverified

glenn-jocher commited on

Update git_describe() for remote dir usage (#2606)
77415a4
unverified

glenn-jocher commited on

PyTorch Hub models default to CUDA:0 if available (#2472)
9b11f0c
unverified

glenn-jocher commited on

Integer printout (#2450)
f01f322
unverified

glenn-jocher commited on

FLOPS min stride 32 (#2276)
83dc1b4
unverified

xiaowo1996 commited on

Update torch_utils.py (#1895)
ffef771
unverified

glenn-jocher commited on

Add colorstr() (#1887)
6ab5895
unverified

glenn-jocher commited on

YOLOv5 v4.0 Release (#1837)
69be8e7
unverified

glenn-jocher commited on

Torch CUDA synchronize update (#1826)
9f5a18b
unverified

glenn-jocher commited on

Display correct CUDA devices (#1776)
c0ffcdf
unverified

glenn-jocher commited on

Automatic m.half() profile on x.half()
3004fb5
unverified

glenn-jocher commited on

Update torch_utils.py
0bd9c48
unverified

glenn-jocher commited on

Input channel yaml['ch'] addition (#1741)
394d1c8
unverified

glenn-jocher commited on

Update loss criteria constructor (#1711)
8bc0027
unverified

glenn-jocher commited on

Fix torch multi-GPU --device error (#1701)
035ac82
unverified

Nanobit glenn-jocher commited on

Profile() feature addition (#1673)
ada90e3
unverified

glenn-jocher commited on

Increase FLOPS robustness (#1608)
8918e63
unverified

glenn-jocher commited on

FLOPS computation device bug fix (#1447)
05a955a
unverified

yujun glenn-jocher commited on

Improved FLOPS computation (#1398)
9f27902

glenn-jocher commited on

Improved FLOPS computation (#1398)
d8f5fcf
unverified

glenn-jocher commited on

Utils reorganization (#1392)
fe341fa
unverified

glenn-jocher commited on

PyTorch version to screen and cleanup (#1325)
19e2482
unverified

glenn-jocher commited on

PyTorch 1.7.0 Compatibility Updates (#1233)
c8c5ef3
unverified

glenn-jocher commited on

Simplified inference (#1153)
3b57cb5
unverified

glenn-jocher commited on

torchvision nms bug fix
124f0e8

glenn-jocher commited on

classifier, export, torch seed updates
883924d

glenn-jocher commited on

.fuse() gradient introduction bug fix
89655a8

glenn-jocher commited on

add support for DWConv fuse operation (#898)
c5969f7
unverified

tkianai commited on

Update torch_utils.py (#751)
26c3b11
unverified

lesscomfortable commited on

Fix redundant outputs via Logging in DDP training (#500)
4949401
unverified

Nanobit glenn-jocher commited on

Single-source training (#680)
3c6e2f7
unverified

glenn-jocher commited on

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

goodfellowliu commited on

update yolo.py TTA flexibility and extensibility (#506)
1d17b9a
unverified

glenn-jocher commited on

v2.0 Release (#491)
9da56b6
unverified

glenn-jocher commited on

update torch_utils.py to FP16 EMA
41ff83a

glenn-jocher commited on

multi-gpu ckpt filesize bug fix #253
a586751

glenn-jocher commited on

--resume EMA fix #292
24c5a94

glenn-jocher commited on