Felix YouandGlenn Jocher c45f9f678d Make select_device() robust to batch_size=-1 (#5940) ...

* Find out a bug. When set batch_size = -1 to use the autobatch.

reproduce:

* Fix type conflict

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>

2021-12-10 15:27:20 +01:00

..

aws

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

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

flask_rest_api

precommit: isort (#5493)

2021-11-04 17:24:25 +01: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

Update wandb_utils.py (#5908)

2021-12-07 16:09:11 +01:00

__init__.py

Absolute '/content/sample_data' (#5922)

2021-12-08 17:31:31 +01:00

activations.py

Update TorchScript suffix to *.torchscript (#5856)

2021-12-02 16:06:45 +01:00

augmentations.py

LOGGER consolidation (#5569)

2021-11-08 16:32:15 +01:00

autoanchor.py

AutoAnchor and AutoBatch LOGGER (#5635)

2021-11-13 15:40:18 +01:00

autobatch.py

AutoAnchor and AutoBatch LOGGER (#5635)

2021-11-13 15:40:18 +01:00

callbacks.py

Optimised Callback Class to Reduce Code and Fix Errors (#4688)

2021-09-07 18:32:15 +02:00

datasets.py

Update nl after cutout() (#5873)

2021-12-03 15:28:14 +01:00

downloads.py

Update autodownload fallbacks to v6.0 assets (#5177)

2021-10-13 19:58:38 -07:00

general.py

AutoShape() models as DetectMultiBackend() instances (#5845)

2021-12-04 15:00:07 +01:00

loss.py

Fix float zeros format (#5491)

2021-11-03 23:36:53 +01:00

metrics.py

Explicitly compute TP, FP in val.py (#5727)

2021-11-20 01:04:56 +01:00

plots.py

Revert "Update plot_lr_scheduler() (#5864)" (#5920)

2021-12-08 16:46:24 +01:00

torch_utils.py

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

2021-12-10 15:27:20 +01:00