xdobro23/YOLOv8-16bit
1
0
Fork 0
Code Issues Pull Requests 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
6 changed files with 120 additions and 98 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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
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
freeze: None # (int | list, optional) freeze first n layers, or freeze list of layer indices during training
# Segmentation
overlap_mask: True # (bool) masks should overlap during training (segment train only)
mask_ratio: 4 # (int) mask downsample ratio (segment train only)