Skip to content

Commit 02abf0e

Browse files
authored
Update how-to-auto-train-image-models.md
1 parent 1135d90 commit 02abf0e

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

articles/machine-learning/how-to-auto-train-image-models.md

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -535,9 +535,9 @@ In general, deep learning model performance can often improve with more data. Da
535535

536536
Currently the augmentations defined above are applied by default for an Automated ML for image job. To provide control over augmentations, Automated ML for images exposes below two flags to turn-off certain augmentations. Currently, these flags are only supported for object detection and instance segmentation tasks.
537537
1. **apply_mosaic_for_yolo:** This flag is only specific to Yolo model. Setting it to False turns off the mosaic data augmentation which is applied at the training time.
538-
2. **apply_automl_train_augmentations:** Setting this flag to false turns off the augmentation applied during training time for the object detection and instance segmentation models. For augmentations, see the details in the table above.
539-
- For non-yolo object detection model and instance segmentation models, this flag turns off only the first three augmentations i.e., *Random crop around bounding boxes, expand, horizontal flip*. The normalization and resize augmentations are still applied regardless of this flag.
540-
- For Yolo model, this flag turns off the random affine and horizontal flip augmentations.
538+
2. **apply_automl_train_augmentations:** Setting this flag to false turns off the augmentation applied during training time for the object detection and instance segmentation models. For augmentations, see the details in the table above.
539+
- For non-yolo object detection model and instance segmentation models, this flag turns off only the first three augmentations i.e., *Random crop around bounding boxes, expand, horizontal flip*. The normalization and resize augmentations are still applied regardless of this flag.
540+
- For Yolo model, this flag turns off the random affine and horizontal flip augmentations.
541541

542542
These two flags are supported via *advanced_settings* under *training_parameters* and can be controlled in the following way.
543543

@@ -561,8 +561,6 @@ training_parameters:
561561
```
562562
In our experiments, we found that these augmentations help the model to generalize better. Therefore, when these augmentations are switched off, we recommend the users to combine them with other offline augmentations to get better results.
563563

564-
---
565-
566564
## Incremental training (optional)
567565

568566
Once the training run is done, you have the option to further train the model by loading the trained model checkpoint. You can either use the same dataset or a different one for incremental training.

0 commit comments

Comments
 (0)