yolov5 / utils /torch_utils.py

Commit History

EarlyStopper updates (#4679)
5487451
unverified

glenn-jocher commited on

Add EarlyStopping feature (#4576)
93cc015
unverified

glenn-jocher commited on

Remove DDP process group timeout (#4422)
19d03a9
unverified

glenn-jocher commited on

Standardize headers and docstrings (#4417)
24bea5e
unverified

glenn-jocher commited on

Fixed logging level in distributed mode (#4284)
771ac6c
unverified

imyhxy fkwong commited on

Update `profile()` for CUDA Memory allocation (#4239)
d8f1883
unverified

glenn-jocher commited on

Update profiler (#4236)
9468657
unverified

glenn-jocher commited on

`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