Skip to content

Commit 03515d8

Browse files
committed
table link
1 parent 324627d commit 03515d8

File tree

2 files changed

+15
-1
lines changed

2 files changed

+15
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -190,7 +190,7 @@ The model algorithm is required and is passed in via `model_name` parameter. You
190190

191191
### Supported model algorithms
192192

193-
The following table summarizes the supported models for each computer vision task. See the reference documentation for model specific hyperparameters.
193+
The following table summarizes the supported models for each computer vision task.
194194

195195
Task | Model algorithms | String literal syntax<br> ***`default_model`\**** denoted with \*
196196
---|----------|----------

articles/machine-learning/reference-automl-images-hyperparameters.md

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,20 @@ Learn which hyperparameters are available specifically for computer vision tasks
1919

2020
With support for computer vision tasks, you can control the model algorithm and sweep hyperparameters. These model algorithms and hyperparameters are passed in as the parameter space for the sweep. While many of the hyperparameters exposed are model-agnostic, there are instances where hyperparameters are task-specific or model-specific.
2121

22+
## Model-specific hyperparameters
23+
24+
This table summarizes hyperparameters specific to the `yolov5` algorithm.
25+
26+
| Parameter name | Description | Default |
27+
| ------------- |-------------|----|
28+
| `validation_metric_type` | Metric computation method to use for validation metrics. <br> Must be `none`, `coco`, `voc`, or `coco_voc`. | `voc` |
29+
| `img_size` | Image size for train and validation. <br> Must be a positive integer. <br> <br> *Note: training run may get into CUDA OOM if the size is too big*. | 640 |
30+
| `model_size` | Model size. <br> Must be `small`, `medium`, `large`, or `xlarge`. <br><br> *Note: training run may get into CUDA OOM if the model size is too big*. | `medium` |
31+
| `multi_scale` | Enable multi-scale image by varying image size by +/- 50% <br> Must be 0 or 1. <br> <br> *Note: training run may get into CUDA OOM if no sufficient GPU memory*. | 0 |
32+
| `box_score_thresh` | During inference, only return proposals with a score greater than `box_score_thresh`. The score is the multiplication of the objectness score and classification probability. <br> Must be a float in the range [0, 1]. | 0.1 |
33+
| `box_iou_thresh` | IoU threshold used during inference in non-maximum suppression post processing. <br> Must be a float in the range [0, 1]. | 0.5 |
34+
35+
2236
## Model agnostic hyperparameters
2337

2438
The following table describes the hyperparameters that are model agnostic.

0 commit comments

Comments
 (0)