New dataset fraction=1.0 argument (#2860)

This commit is contained in:
Glenn Jocher
2023-05-28 02:13:46 +02:00
committed by GitHub
parent 61fa5efe6d
commit 0bdd4ad379
7 changed files with 16 additions and 5 deletions

View File

@ -82,6 +82,7 @@ task.
| `close_mosaic` | `0` | (int) disable mosaic augmentation for final epochs |
| `resume` | `False` | resume training from last checkpoint |
| `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) |
| `lr0` | `0.01` | initial learning rate (i.e. SGD=1E-2, Adam=1E-3) |
| `lrf` | `0.01` | final learning rate (lr0 * lrf) |
| `momentum` | `0.937` | SGD momentum/Adam beta1 |