From 2193ca1658007013c25cb427b1ddfbdfeb3c3829 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 4 Jun 2023 22:38:25 +0200 Subject: [PATCH] `ultralytics 8.0.113` Adamax, NAdam, RAdam optimizers (#3004) --- docs/modes/train.md | 2 +- docs/usage/cfg.md | 2 +- ultralytics/__init__.py | 2 +- ultralytics/yolo/cfg/default.yaml | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/docs/modes/train.md b/docs/modes/train.md index 4492373..d905084 100644 --- a/docs/modes/train.md +++ b/docs/modes/train.md @@ -72,7 +72,7 @@ task. | `name` | `None` | experiment name | | `exist_ok` | `False` | whether to overwrite existing experiment | | `pretrained` | `False` | whether to use a pretrained model | -| `optimizer` | `'auto'` | optimizer to use, choices=[SGD, Adam, Adamax, AdamW, NAdam, RAdam, RMSProp, auto] | +| `optimizer` | `'SGD'` | optimizer to use, choices=[SGD, Adam, Adamax, AdamW, NAdam, RAdam, RMSProp, auto] | | `verbose` | `False` | whether to print verbose output | | `seed` | `0` | random seed for reproducibility | | `deterministic` | `True` | whether to enable deterministic mode | diff --git a/docs/usage/cfg.md b/docs/usage/cfg.md index b7ddb49..f6dc187 100644 --- a/docs/usage/cfg.md +++ b/docs/usage/cfg.md @@ -94,7 +94,7 @@ The training settings for YOLO models encompass various hyperparameters and conf | `name` | `None` | experiment name | | `exist_ok` | `False` | whether to overwrite existing experiment | | `pretrained` | `False` | whether to use a pretrained model | -| `optimizer` | `'auto'` | optimizer to use, choices=[SGD, Adam, Adamax, AdamW, NAdam, RAdam, RMSProp, auto] | +| `optimizer` | `'SGD'` | optimizer to use, choices=[SGD, Adam, Adamax, AdamW, NAdam, RAdam, RMSProp, auto] | | `verbose` | `False` | whether to print verbose output | | `seed` | `0` | random seed for reproducibility | | `deterministic` | `True` | whether to enable deterministic mode | diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py index accf0f5..86f7d7c 100644 --- a/ultralytics/__init__.py +++ b/ultralytics/__init__.py @@ -1,6 +1,6 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license -__version__ = '8.0.112' +__version__ = '8.0.113' from ultralytics.hub import start from ultralytics.vit.rtdetr import RTDETR diff --git a/ultralytics/yolo/cfg/default.yaml b/ultralytics/yolo/cfg/default.yaml index 35be2e9..cd73148 100644 --- a/ultralytics/yolo/cfg/default.yaml +++ b/ultralytics/yolo/cfg/default.yaml @@ -20,7 +20,7 @@ project: # project name name: # experiment name, results saved to 'project/name' directory exist_ok: False # whether to overwrite existing experiment pretrained: False # whether to use a pretrained model -optimizer: auto # optimizer to use, choices=[SGD, Adam, Adamax, AdamW, NAdam, RAdam, RMSProp, auto] +optimizer: SGD # optimizer to use, choices=[SGD, Adam, Adamax, AdamW, NAdam, RAdam, RMSProp, auto] verbose: True # whether to print verbose output seed: 0 # random seed for reproducibility deterministic: True # whether to enable deterministic mode