yolov5 / utils /torch_utils.py

Commit History

Fix2 `select_device()` for Multi-GPU (#6461)
d8b5beb
unverified

glenn-jocher commited on

Fix `select_device()` for Multi-GPU (#6434)
856d4e5
unverified

glenn-jocher commited on

Reuse `de_parallel()` rather than `is_parallel()` (#6354)
4e841b9
unverified

imyhxy commited on

`select_device()` cleanup (#6302)
436ffc4
unverified

glenn-jocher commited on

Fixing bug multi-gpu training (#6299)
a1a9c68
unverified

Henry commited on

Fix `device` count check (#6290)
e7bf382
unverified

glenn-jocher commited on

Make `select_device()` robust to `batch_size=-1` (#5940)
c45f9f6
unverified

Felix You glenn-jocher commited on

Add `notebook_init()` to utils/__init__.py (#5488)
7ebb5e5
unverified

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

New `DetectMultiBackend()` class (#5549)
3883261
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

Fix float zeros format (#5491)
5866646
unverified

glenn-jocher commited on

Update torch_utils.py import `LOGGER` (#5483)
84a8099
unverified

glenn-jocher commited on

Improve GPU name (#5478)
df30426
unverified

glenn-jocher commited on

Fix MixConv2d() remove shortcut + apply depthwise (#5410)
5d4258f
unverified

glenn-jocher commited on

Add pre-commit CI actions (#4982)
ed887b5
unverified

Jirka Borovec glenn-jocher commited on

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

glenn-jocher commited on

More informative `EarlyStopping()` message (#5303)
441b47c
unverified

glenn-jocher commited on

Consolidate `init_seeds()` (#4849)
84bfa89
unverified

glenn-jocher commited on

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