| --- |
| description: Learn how to train YOLO models for segmentation tasks with Ultralytics. Explore the SegmentationTrainer class and its functionalities. |
| keywords: YOLO, segmentation, train, Ultralytics, SegmentationTrainer, Python, machine learning, deep learning, tutorials |
| --- |
| |
| # Reference for `ultralytics/models/yolo/segment/train.py` |
|
|
| !!! Note |
|
|
| This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/segment/train.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/segment/train.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/models/yolo/segment/train.py) 🛠️. Thank you 🙏! |
| |
| <br> |
|
|
| ## ::: ultralytics.models.yolo.segment.train.SegmentationTrainer |
|
|
| <br><br> |
|
|
