Glenn Jocher 9b11f0c58b PyTorch Hub models default to CUDA:0 if available (#2472) ...

* PyTorch Hub models default to CUDA:0 if available

* device as string bug fix

2021-03-14 23:16:17 -07:00

..

aws

GCP sudo docker userdata.sh (#2393)

2021-03-07 20:21:49 -08:00

google_app_engine

adding the configuration to deploy yolov5 in in app engine (#964)

2020-09-22 11:48:44 -07:00

wandb_logging

Start setup for improved W&B integration (#1948)

2021-02-01 21:38:41 -08:00

__init__.py

initial commit

2020-05-29 17:04:54 -07:00

activations.py

nn.SiLU() citation correction (#1713)

2021-01-06 20:58:41 -08:00

autoanchor.py

Scipy kmeans-robust autoanchor update (#2470)

2021-03-14 21:58:12 -07:00

datasets.py

PyTorch Hub models default to CUDA:0 if available (#2472)

2021-03-14 23:16:17 -07:00

general.py

PyTorch Hub models default to CUDA:0 if available (#2472)

2021-03-14 23:16:17 -07:00

google_utils.py

Update google_utils.py (#2017)

2021-01-22 15:08:42 -08:00

loss.py

Robust objectness loss balancing (#2256)

2021-02-20 11:19:01 -08:00

metrics.py

Confusion matrix background axis swap (#2114)

2021-03-02 16:09:52 -08:00

plots.py

labels.jpg class names (#2454)

2021-03-12 22:15:41 -08:00

torch_utils.py

PyTorch Hub models default to CUDA:0 if available (#2472)

2021-03-14 23:16:17 -07:00