Glenn Jocher ec1d8496ba Improved model+EMA checkpointing (#2292) ...

* Enhanced model+EMA checkpointing

* update

* bug fix

* bug fix 2

* always save optimizer

* ema half

* remove model.float()

* model half

* carry ema/model in fp32

* rm model.float()

* both to float always

* cleanup

* cleanup

2021-02-24 18:26:46 -08:00

..

aws

Amazon AWS EC2 startup and re-startup scripts (#2282)

2021-02-24 01:43:59 -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

Update autoanchor.py with yaml.SafeLoader (#1971)

2021-01-18 10:47:42 -08:00

datasets.py

LoadStreams() frame loss bug fix (#2222)

2021-02-15 11:02:20 -08:00

general.py

Improved model+EMA checkpointing (#2292)

2021-02-24 18:26:46 -08: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

Metric-Confidence plots feature addition (#2057)

2021-01-27 17:10:53 -08:00

plots.py

Update inference default to multi_label=False (#2252)

2021-02-19 15:39:09 -08:00

torch_utils.py

FLOPS min stride 32 (#2276)

2021-02-23 09:38:56 -08:00