Glenn Jocher dd28df98c2 Avoid FP64 ops for MPS support in train.py (#8511) ...

Avoid FP64 ops for MPS support

Resolves https://github.com/ultralytics/yolov5/pull/7878#issuecomment-1177952614

2022-07-07 20:36:23 +02:00

..

aws

Fix yaml.safe_load() ignore emoji errors (#5060)

2021-10-05 13:41:52 -07:00

docker

Docker GPU pip install opencv-python<4.6.0.66 (#8164)

2022-06-09 22:45:49 +02:00

flask_rest_api

Code refactor (#7923)

2022-05-22 15:24:08 +02:00

google_app_engine

Bump pip from 19.2 to 21.1 in /utils/google_app_engine (#5661)

2021-11-15 20:10:29 +01:00

loggers

Refactor collections and fstrings (#7821)

2022-05-15 16:38:26 +02:00

__init__.py

Update bytes to GB with bitshift (#6886)

2022-03-07 19:26:37 +01:00

activations.py

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

2022-04-04 22:47:00 +02:00

augmentations.py

precommit: yapf (#5494)

2022-03-31 16:52:34 +02:00

autoanchor.py

Refactor collections and fstrings (#7821)

2022-05-15 16:38:26 +02:00

autobatch.py

AutoBatch checks against failed solutions (#8159)

2022-06-09 17:15:49 +02:00

benchmarks.py

Add File Size (MB) column to benchmarks (#8359)

2022-06-27 17:46:49 +02:00

callbacks.py

Simplify callbacks.py return (#7333)

2022-04-07 16:15:01 +02:00

dataloaders.py

Update cache comments (#8414)

2022-06-30 16:19:22 +02:00

downloads.py

Pin downloads to release version (#7790)

2022-05-13 00:24:53 +02:00

general.py

Avoid FP64 ops for MPS support in train.py (#8511)

2022-07-07 20:36:23 +02:00

loss.py

Fix torch long to float tensor on HUB macOS (#8067)

2022-06-01 17:34:46 +02:00

metrics.py

Bug fix mAP0.5-0.95 (#6787)

2022-05-20 11:33:10 +02:00

plots.py

Add @threaded decorator (#7813)

2022-05-14 16:12:08 +02:00

torch_utils.py

Do not prefer Apple MPS (#8446)

2022-07-02 18:35:45 +02:00