Skip to content

Commit 812834f

Browse files
nvijayraniarjaincc
andauthored
Update articles/machine-learning/how-to-auto-train-image-models.md
Co-authored-by: Rupal jain <[email protected]>
1 parent 54956bb commit 812834f

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

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

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -535,7 +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 job. To provide control over augmentations, Automated ML expose flags to turn-off certain augmentations. Currently, these flags are only supported for object detection and instance segmentation tasks. We expose following two flags to turn-off the augmentations.
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:** This flag turns off the augmentation applied at the train time to the object detection and instance segmentation models. For augmentations, please see the details in the table above. Please note that for non-yolo object detection model and instance segmentation model, this flag turns off only first three augmentations i.e. *Random crop around bounding boxes, expand, horizontal flip*. The normalization and resize augmentation is still applied regardless of this flag.
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.
539541
For Yolo model, this flag turns off the random affine and horizontal flip augmentation.
540542

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

0 commit comments

Comments
 (0)