Régler un modèle Object Detection - TensorFlow - Amazon SageMaker AI

Les traductions sont fournies par des outils de traduction automatique. En cas de conflit entre le contenu d'une traduction et celui de la version originale en anglais, la version anglaise prévaudra.

Régler un modèle Object Detection - TensorFlow

Le réglage de modèle automatique, ou réglage d'hyperparamètre, détecte la meilleure version d'un modèle en exécutant plusieurs tâches qui testent une plage d'hyperparamètres sur votre jeu de données. Vous choisissez les hyperparamètres réglables, une plage de valeurs pour chacun d'eux et une métrique d'objectif. Vous choisissez la métrique d'objectif parmi les métriques que calcule l'algorithme. Le réglage de modèle automatique recherche parmi les hyperparamètres choisis la combinaison de valeurs qui produira un modèle permettant d'optimiser la métrique d'objectif.

Pour plus d'informations sur le réglage de modèle, consultez Réglage automatique du modèle grâce à l' SageMaker IA.

Métriques calculées par l'algorithme Object Detection - TensorFlow

Reportez-vous au tableau suivant pour savoir quelles mesures sont calculées par l'algorithme Object Detection - TensorFlow.

Nom de la métrique Description Orientation de l'optimisation Motif Regex
validation:localization_loss

La perte de localisation pour la prédiction des boîtes.

Réduire

Val_localization=([0-9\\.]+)

Hyperparamètres réglables pour Object Detection - TensorFlow

Personnalisez un modèle de détection d'objet avec les hyperparamètres suivants. Les hyperparamètres qui ont le plus d'impact sur la métrique d'objectif de détection d'objet sont : batch_size, learning_rate et optimizer. Réglez les hyperparamètres associés à l'optimiseur, tels que momentum, regularizers_l2, beta_1, beta_2 et eps, en fonction de l'optimizer sélectionné. Par exemple, utilisez beta_1 et beta_2 uniquement si adam = optimizer.

Pour plus d'informations sur les hyperparamètres qui sont utilisés pour chaque optimizer, consultez Hyperparamètres pour Object Detection - TensorFlow.

Nom du paramètre Type de paramètre Plages recommandées
batch_size

IntegerParameterRanges

Valeur min. : 8, Valeur max. : 512

beta_1

ContinuousParameterRanges

Valeur min. : 1e-6, Valeur max. : 0,999

beta_2

ContinuousParameterRanges

Valeur min. : 1e-6, Valeur max. : 0,999

eps

ContinuousParameterRanges

Valeur min. : 1e-8, Valeur max. : 1

learning_rate

ContinuousParameterRanges

Valeur min. : 1e-6, Valeur max. : 0,5

momentum

ContinuousParameterRanges

Valeur min. : 0, Valeur max. : 0,999

optimizer

CategoricalParameterRanges

['sgd', ‘adam’, ‘rmsprop’, 'nesterov', 'adagrad', 'adadelta']

regularizers_l2

ContinuousParameterRanges

Valeur min. : 0, Valeur max. : 0,999

train_only_on_top_layer

CategoricalParameterRanges

['True', 'False']

initial_accumulator_value

CategoricalParameterRanges

Valeur min. : 0, Valeur max. : 0,999