xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 21:44:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

ultralytics 8.0.154 add freeze training argument (#4329)

Browse Source
This commit is contained in:
Glenn Jocher 2023-08-14 00:18:32 +02:00 committed by GitHub
parent 9f6d48d3cf
commit d47718c367
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
6 changed files with 120 additions and 98 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

95
docs/modes/train.md
View File

@ -140,53 +140,54 @@ Remember that checkpoints are saved at the end of every epoch by default, or at
Training settings for YOLO models refer to the various hyperparameters and configurations used to train the model on a dataset. These settings can affect the model's performance, speed, and accuracy. Some common YOLO training settings include the batch size, learning rate, momentum, and weight decay. Other factors that may affect the training process include the choice of optimizer, the choice of loss function, and the size and composition of the training dataset. It is important to carefully tune and experiment with these settings to achieve the best possible performance for a given task. Training settings for YOLO models refer to the various hyperparameters and configurations used to train the model on a dataset. These settings can affect the model's performance, speed, and accuracy. Some common YOLO training settings include the batch size, learning rate, momentum, and weight decay. Other factors that may affect the training process include the choice of optimizer, the choice of loss function, and the size and composition of the training dataset. It is important to carefully tune and experiment with these settings to achieve the best possible performance for a given task.
| Key | Value | Description | | Key | Value | Description |
|-------------------|----------|-----------------------------------------------------------------------------------| |-------------------|----------|------------------------------------------------------------------------------------------------|
| `model` | `None` | path to model file, i.e. yolov8n.pt, yolov8n.yaml | | `model` | `None` | path to model file, i.e. yolov8n.pt, yolov8n.yaml |
| `data` | `None` | path to data file, i.e. coco128.yaml | | `data` | `None` | path to data file, i.e. coco128.yaml |
| `epochs` | `100` | number of epochs to train for | | `epochs` | `100` | number of epochs to train for |
| `patience` | `50` | epochs to wait for no observable improvement for early stopping of training | | `patience` | `50` | epochs to wait for no observable improvement for early stopping of training |
| `batch` | `16` | number of images per batch (-1 for AutoBatch) | | `batch` | `16` | number of images per batch (-1 for AutoBatch) |
| `imgsz` | `640` | size of input images as integer | | `imgsz` | `640` | size of input images as integer |
| `save` | `True` | save train checkpoints and predict results | | `save` | `True` | save train checkpoints and predict results |
| `save_period` | `-1` | Save checkpoint every x epochs (disabled if < 1) | | `save_period` | `-1` | Save checkpoint every x epochs (disabled if < 1) |
| `cache` | `False` | True/ram, disk or False. Use cache for data loading | | `cache` | `False` | True/ram, disk or False. Use cache for data loading |
| `device` | `None` | device to run on, i.e. cuda device=0 or device=0,1,2,3 or device=cpu | | `device` | `None` | device to run on, i.e. cuda device=0 or device=0,1,2,3 or device=cpu |
| `workers` | `8` | number of worker threads for data loading (per RANK if DDP) | | `workers` | `8` | number of worker threads for data loading (per RANK if DDP) |
| `project` | `None` | project name | | `project` | `None` | project name |
| `name` | `None` | experiment name | | `name` | `None` | experiment name |
| `exist_ok` | `False` | whether to overwrite existing experiment | | `exist_ok` | `False` | whether to overwrite existing experiment |
| `pretrained` | `False` | whether to use a pretrained model | | `pretrained` | `False` | whether to use a pretrained model |
| `optimizer` | `'auto'` | optimizer to use, choices=[SGD, Adam, Adamax, AdamW, NAdam, RAdam, RMSProp, auto] | | `optimizer` | `'auto'` | optimizer to use, choices=[SGD, Adam, Adamax, AdamW, NAdam, RAdam, RMSProp, auto] |
| `verbose` | `False` | whether to print verbose output | | `verbose` | `False` | whether to print verbose output |
| `seed` | `0` | random seed for reproducibility | | `seed` | `0` | random seed for reproducibility |
| `deterministic` | `True` | whether to enable deterministic mode | | `deterministic` | `True` | whether to enable deterministic mode |
| `single_cls` | `False` | train multi-class data as single-class | | `single_cls` | `False` | train multi-class data as single-class |
| `rect` | `False` | rectangular training with each batch collated for minimum padding | | `rect` | `False` | rectangular training with each batch collated for minimum padding |
| `cos_lr` | `False` | use cosine learning rate scheduler | | `cos_lr` | `False` | use cosine learning rate scheduler |
| `close_mosaic` | `10` | (int) disable mosaic augmentation for final epochs (0 to disable) | | `close_mosaic` | `10` | (int) disable mosaic augmentation for final epochs (0 to disable) |
| `resume` | `False` | resume training from last checkpoint | | `resume` | `False` | resume training from last checkpoint |
| `amp` | `True` | Automatic Mixed Precision (AMP) training, choices=[True, False] | | `amp` | `True` | Automatic Mixed Precision (AMP) training, choices=[True, False] |
| `fraction` | `1.0` | dataset fraction to train on (default is 1.0, all images in train set) | | `fraction` | `1.0` | dataset fraction to train on (default is 1.0, all images in train set) |
| `profile` | `False` | profile ONNX and TensorRT speeds during training for loggers | | `profile` | `False` | profile ONNX and TensorRT speeds during training for loggers |
| `lr0` | `0.01` | initial learning rate (i.e. SGD=1E-2, Adam=1E-3) | | `freeze` | `None` | (int or list, optional) freeze first n layers, or freeze list of layer indices during training |
| `lrf` | `0.01` | final learning rate (lr0 * lrf) | | `lr0` | `0.01` | initial learning rate (i.e. SGD=1E-2, Adam=1E-3) |
| `momentum` | `0.937` | SGD momentum/Adam beta1 | | `lrf` | `0.01` | final learning rate (lr0 * lrf) |
| `weight_decay` | `0.0005` | optimizer weight decay 5e-4 | | `momentum` | `0.937` | SGD momentum/Adam beta1 |
| `warmup_epochs` | `3.0` | warmup epochs (fractions ok) | | `weight_decay` | `0.0005` | optimizer weight decay 5e-4 |
| `warmup_momentum` | `0.8` | warmup initial momentum | | `warmup_epochs` | `3.0` | warmup epochs (fractions ok) |
| `warmup_bias_lr` | `0.1` | warmup initial bias lr | | `warmup_momentum` | `0.8` | warmup initial momentum |
| `box` | `7.5` | box loss gain | | `warmup_bias_lr` | `0.1` | warmup initial bias lr |
| `cls` | `0.5` | cls loss gain (scale with pixels) | | `box` | `7.5` | box loss gain |
| `dfl` | `1.5` | dfl loss gain | | `cls` | `0.5` | cls loss gain (scale with pixels) |
| `pose` | `12.0` | pose loss gain (pose-only) | | `dfl` | `1.5` | dfl loss gain |
| `kobj` | `2.0` | keypoint obj loss gain (pose-only) | | `pose` | `12.0` | pose loss gain (pose-only) |
| `label_smoothing` | `0.0` | label smoothing (fraction) | | `kobj` | `2.0` | keypoint obj loss gain (pose-only) |
| `nbs` | `64` | nominal batch size | | `label_smoothing` | `0.0` | label smoothing (fraction) |
| `overlap_mask` | `True` | masks should overlap during training (segment train only) | | `nbs` | `64` | nominal batch size |
| `mask_ratio` | `4` | mask downsample ratio (segment train only) | | `overlap_mask` | `True` | masks should overlap during training (segment train only) |
| `dropout` | `0.0` | use dropout regularization (classify train only) | | `mask_ratio` | `4` | mask downsample ratio (segment train only) |
| `val` | `True` | validate/test during training | | `dropout` | `0.0` | use dropout regularization (classify train only) |
| `val` | `True` | validate/test during training |
## Logging ## Logging

95
docs/usage/cfg.md
View File

@ -78,53 +78,54 @@ include:
The training settings for YOLO models encompass various hyperparameters and configurations used during the training process. These settings influence the model's performance, speed, and accuracy. Key training settings include batch size, learning rate, momentum, and weight decay. Additionally, the choice of optimizer, loss function, and training dataset composition can impact the training process. Careful tuning and experimentation with these settings are crucial for optimizing performance. The training settings for YOLO models encompass various hyperparameters and configurations used during the training process. These settings influence the model's performance, speed, and accuracy. Key training settings include batch size, learning rate, momentum, and weight decay. Additionally, the choice of optimizer, loss function, and training dataset composition can impact the training process. Careful tuning and experimentation with these settings are crucial for optimizing performance.
| Key | Value | Description | | Key | Value | Description |
|-------------------|----------|-----------------------------------------------------------------------------------| |-------------------|----------|------------------------------------------------------------------------------------------------|
| `model` | `None` | path to model file, i.e. yolov8n.pt, yolov8n.yaml | | `model` | `None` | path to model file, i.e. yolov8n.pt, yolov8n.yaml |
| `data` | `None` | path to data file, i.e. coco128.yaml | | `data` | `None` | path to data file, i.e. coco128.yaml |
| `epochs` | `100` | number of epochs to train for | | `epochs` | `100` | number of epochs to train for |
| `patience` | `50` | epochs to wait for no observable improvement for early stopping of training | | `patience` | `50` | epochs to wait for no observable improvement for early stopping of training |
| `batch` | `16` | number of images per batch (-1 for AutoBatch) | | `batch` | `16` | number of images per batch (-1 for AutoBatch) |
| `imgsz` | `640` | size of input images as integer or w,h | | `imgsz` | `640` | size of input images as integer or w,h |
| `save` | `True` | save train checkpoints and predict results | | `save` | `True` | save train checkpoints and predict results |
| `save_period` | `-1` | Save checkpoint every x epochs (disabled if < 1) | | `save_period` | `-1` | Save checkpoint every x epochs (disabled if < 1) |
| `cache` | `False` | True/ram, disk or False. Use cache for data loading | | `cache` | `False` | True/ram, disk or False. Use cache for data loading |
| `device` | `None` | device to run on, i.e. cuda device=0 or device=0,1,2,3 or device=cpu | | `device` | `None` | device to run on, i.e. cuda device=0 or device=0,1,2,3 or device=cpu |
| `workers` | `8` | number of worker threads for data loading (per RANK if DDP) | | `workers` | `8` | number of worker threads for data loading (per RANK if DDP) |
| `project` | `None` | project name | | `project` | `None` | project name |
| `name` | `None` | experiment name | | `name` | `None` | experiment name |
| `exist_ok` | `False` | whether to overwrite existing experiment | | `exist_ok` | `False` | whether to overwrite existing experiment |
| `pretrained` | `False` | whether to use a pretrained model | | `pretrained` | `False` | whether to use a pretrained model |
| `optimizer` | `'auto'` | optimizer to use, choices=[SGD, Adam, Adamax, AdamW, NAdam, RAdam, RMSProp, auto] | | `optimizer` | `'auto'` | optimizer to use, choices=[SGD, Adam, Adamax, AdamW, NAdam, RAdam, RMSProp, auto] |
| `verbose` | `False` | whether to print verbose output | | `verbose` | `False` | whether to print verbose output |
| `seed` | `0` | random seed for reproducibility | | `seed` | `0` | random seed for reproducibility |
| `deterministic` | `True` | whether to enable deterministic mode | | `deterministic` | `True` | whether to enable deterministic mode |
| `single_cls` | `False` | train multi-class data as single-class | | `single_cls` | `False` | train multi-class data as single-class |
| `rect` | `False` | rectangular training with each batch collated for minimum padding | | `rect` | `False` | rectangular training with each batch collated for minimum padding |
| `cos_lr` | `False` | use cosine learning rate scheduler | | `cos_lr` | `False` | use cosine learning rate scheduler |
| `close_mosaic` | `10` | (int) disable mosaic augmentation for final epochs (0 to disable) | | `close_mosaic` | `10` | (int) disable mosaic augmentation for final epochs (0 to disable) |
| `resume` | `False` | resume training from last checkpoint | | `resume` | `False` | resume training from last checkpoint |
| `amp` | `True` | Automatic Mixed Precision (AMP) training, choices=[True, False] | | `amp` | `True` | Automatic Mixed Precision (AMP) training, choices=[True, False] |
| `fraction` | `1.0` | dataset fraction to train on (default is 1.0, all images in train set) | | `fraction` | `1.0` | dataset fraction to train on (default is 1.0, all images in train set) |
| `profile` | `False` | profile ONNX and TensorRT speeds during training for loggers | | `profile` | `False` | profile ONNX and TensorRT speeds during training for loggers |
| `lr0` | `0.01` | initial learning rate (i.e. SGD=1E-2, Adam=1E-3) | | `freeze` | `None` | (int or list, optional) freeze first n layers, or freeze list of layer indices during training |
| `lrf` | `0.01` | final learning rate (lr0 * lrf) | | `lr0` | `0.01` | initial learning rate (i.e. SGD=1E-2, Adam=1E-3) |
| `momentum` | `0.937` | SGD momentum/Adam beta1 | | `lrf` | `0.01` | final learning rate (lr0 * lrf) |
| `weight_decay` | `0.0005` | optimizer weight decay 5e-4 | | `momentum` | `0.937` | SGD momentum/Adam beta1 |
| `warmup_epochs` | `3.0` | warmup epochs (fractions ok) | | `weight_decay` | `0.0005` | optimizer weight decay 5e-4 |
| `warmup_momentum` | `0.8` | warmup initial momentum | | `warmup_epochs` | `3.0` | warmup epochs (fractions ok) |
| `warmup_bias_lr` | `0.1` | warmup initial bias lr | | `warmup_momentum` | `0.8` | warmup initial momentum |
| `box` | `7.5` | box loss gain | | `warmup_bias_lr` | `0.1` | warmup initial bias lr |
| `cls` | `0.5` | cls loss gain (scale with pixels) | | `box` | `7.5` | box loss gain |
| `dfl` | `1.5` | dfl loss gain | | `cls` | `0.5` | cls loss gain (scale with pixels) |
| `pose` | `12.0` | pose loss gain (pose-only) | | `dfl` | `1.5` | dfl loss gain |
| `kobj` | `2.0` | keypoint obj loss gain (pose-only) | | `pose` | `12.0` | pose loss gain (pose-only) |
| `label_smoothing` | `0.0` | label smoothing (fraction) | | `kobj` | `2.0` | keypoint obj loss gain (pose-only) |
| `nbs` | `64` | nominal batch size | | `label_smoothing` | `0.0` | label smoothing (fraction) |
| `overlap_mask` | `True` | masks should overlap during training (segment train only) | | `nbs` | `64` | nominal batch size |
| `mask_ratio` | `4` | mask downsample ratio (segment train only) | | `overlap_mask` | `True` | masks should overlap during training (segment train only) |
| `dropout` | `0.0` | use dropout regularization (classify train only) | | `mask_ratio` | `4` | mask downsample ratio (segment train only) |
| `val` | `True` | validate/test during training | | `dropout` | `0.0` | use dropout regularization (classify train only) |
| `val` | `True` | validate/test during training |
[Train Guide](../modes/train.md){ .md-button .md-button--primary} [Train Guide](../modes/train.md){ .md-button .md-button--primary}

2
ultralytics/__init__.py
View File

@ -1,6 +1,6 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics YOLO 🚀, AGPL-3.0 license
__version__ = '8.0.153' __version__ = '8.0.154'
from ultralytics.hub import start from ultralytics.hub import start
from ultralytics.models import RTDETR, SAM, YOLO from ultralytics.models import RTDETR, SAM, YOLO

1
ultralytics/cfg/default.yaml
View File

@ -32,6 +32,7 @@ resume: False # (bool) resume training from last checkpoint
amp: True # (bool) Automatic Mixed Precision (AMP) training, choices=[True, False], True runs AMP check amp: True # (bool) Automatic Mixed Precision (AMP) training, choices=[True, False], True runs AMP check
fraction: 1.0 # (float) dataset fraction to train on (default is 1.0, all images in train set) fraction: 1.0 # (float) dataset fraction to train on (default is 1.0, all images in train set)
profile: False # (bool) profile ONNX and TensorRT speeds during training for loggers profile: False # (bool) profile ONNX and TensorRT speeds during training for loggers
freeze: None # (int | list, optional) freeze first n layers, or freeze list of layer indices during training
# Segmentation # Segmentation
overlap_mask: True # (bool) masks should overlap during training (segment train only) overlap_mask: True # (bool) masks should overlap during training (segment train only)
mask_ratio: 4 # (int) mask downsample ratio (segment train only) mask_ratio: 4 # (int) mask downsample ratio (segment train only)

20
ultralytics/engine/trainer.py
View File

@ -207,11 +207,28 @@ class BaseTrainer:
""" """
Builds dataloaders and optimizer on correct rank process. Builds dataloaders and optimizer on correct rank process.
""" """
# Model # Model
self.run_callbacks('on_pretrain_routine_start') self.run_callbacks('on_pretrain_routine_start')
ckpt = self.setup_model() ckpt = self.setup_model()
self.model = self.model.to(self.device) self.model = self.model.to(self.device)
self.set_model_attributes() self.set_model_attributes()
# Freeze layers
freeze_list = self.args.freeze if isinstance(
self.args.freeze, list) else range(self.args.freeze) if isinstance(self.args.freeze, int) else []
always_freeze_names = ['.dfl'] # always freeze these layers
freeze_layer_names = [f'model.{x}.' for x in freeze_list] + always_freeze_names
for k, v in self.model.named_parameters():
# v.register_hook(lambda x: torch.nan_to_num(x)) # NaN to 0 (commented for erratic training results)
if any(x in k for x in freeze_layer_names):
LOGGER.info(f"Freezing layer '{k}'")
v.requires_grad = False
elif not v.requires_grad:
LOGGER.info(f"WARNING ⚠️ setting 'requires_grad=True' for frozen layer '{k}'. "
'See ultralytics.engine.trainer for customization of frozen layers.')
v.requires_grad = True
# Check AMP # Check AMP
self.amp = torch.tensor(self.args.amp).to(self.device) # True or False self.amp = torch.tensor(self.args.amp).to(self.device) # True or False
if self.amp and RANK in (-1, 0): # Single-GPU and DDP if self.amp and RANK in (-1, 0): # Single-GPU and DDP
@ -224,9 +241,11 @@ class BaseTrainer:
self.scaler = amp.GradScaler(enabled=self.amp) self.scaler = amp.GradScaler(enabled=self.amp)
if world_size > 1: if world_size > 1:
self.model = DDP(self.model, device_ids=[RANK]) self.model = DDP(self.model, device_ids=[RANK])
# Check imgsz # Check imgsz
gs = max(int(self.model.stride.max() if hasattr(self.model, 'stride') else 32), 32) # grid size (max stride) gs = max(int(self.model.stride.max() if hasattr(self.model, 'stride') else 32), 32) # grid size (max stride)
self.args.imgsz = check_imgsz(self.args.imgsz, stride=gs, floor=gs, max_dim=1) self.args.imgsz = check_imgsz(self.args.imgsz, stride=gs, floor=gs, max_dim=1)
# Batch size # Batch size
if self.batch_size == -1: if self.batch_size == -1:
if RANK == -1: # single-GPU only, estimate best batch size if RANK == -1: # single-GPU only, estimate best batch size
@ -272,7 +291,6 @@ class BaseTrainer:
"""Train completed, evaluate and plot if specified by arguments.""" """Train completed, evaluate and plot if specified by arguments."""
if world_size > 1: if world_size > 1:
self._setup_ddp(world_size) self._setup_ddp(world_size)
self._setup_train(world_size) self._setup_train(world_size)
self.epoch_time = None self.epoch_time = None

5
ultralytics/nn/modules/conv.py
View File

@ -142,8 +142,9 @@ class GhostConv(nn.Module):
class RepConv(nn.Module): class RepConv(nn.Module):
"""RepConv is a basic rep-style block, including training and deploy status """
This code is based on https://github.com/DingXiaoH/RepVGG/blob/main/repvgg.py RepConv is a basic rep-style block, including training and deploy status. This module is used in RT-DETR.
Based on https://github.com/DingXiaoH/RepVGG/blob/main/repvgg.py
""" """
default_act = nn.SiLU() # default activation default_act = nn.SiLU() # default activation