yolov5 / utils /torch_utils.py

Commit History

YOLOv5 Apple Metal Performance Shader (MPS) support (#7878)
c215878
unverified

glenn-jocher commited on

Added Windows cmd to count GPU devices (#7891)
f43cd53
unverified

xylieong glenn-jocher commited on

Refactor collections and fstrings (#7821)
f000714
unverified

glenn-jocher commited on

Add Python version output (#7814)
3356f26
unverified

glenn-jocher commited on

[pre-commit.ci] pre-commit suggestions (#7279)
7882950
unverified

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

precommit: yapf (#5494)
c3d5ac1
unverified

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

Model summary `pathlib` fix (#7104)
6134ec5
unverified

glenn-jocher commited on

Update model summary to display model name (#7101)
a600bae
unverified

glenn-jocher commited on

Move `git_describe()` to general.py (#6918)
6dd82c0
unverified

glenn-jocher commited on

Update bytes to GB with bitshift (#6886)
e6e36aa
unverified

glenn-jocher commited on

Update EMA decay `tau` (#6769)
0f81991
unverified

glenn-jocher commited on

Suppress `torch` AMP-CPU warnings (#6706)
4de8b24
unverified

glenn-jocher commited on

Fix 6 Flake8 issues (#6541)
cba4303
unverified

Jirka Borovec glenn-jocher commited on

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