Hiperparámetros de detección de objetos - TensorFlow - Amazon SageMaker AI

Hiperparámetros de detección de objetos - TensorFlow

Los hiperparámetros son parámetros que se establecen antes de que un modelo de machine learning comience a aprender. Los siguientes hiperparámetros son compatibles con el algoritmo de detección de objetos - TensorFlow integrado en Amazon SageMaker AI. Para obtener más información sobre el ajuste de hiperparámetros, consulte Ajuste de un modelo de detección de objetos - TensorFlow.

Nombre del parámetro Descripción
batch_size

El tamaño del lote para la capacitación.

Valores válidos: número entero positivo.

Valor predeterminado: 3.
beta_1

La versión beta1 del optimizador "adam". Representa la tasa de degradación exponencial para las estimaciones del primer momento. No se tiene en cuenta para otros optimizadores.

Valores válidos: flotante, con el rango [0.0, 1.0].

Valor predeterminado: 0.9.
beta_2

La versión beta2 del optimizador "adam". Representa la tasa de degradación exponencial para las estimaciones del segundo momento. No se tiene en cuenta para otros optimizadores.

Valores válidos: flotante, con el rango [0.0, 1.0].

Valor predeterminado: 0.999.
early_stopping

Se establece en "True" a fin de usar una lógica de detención temprana durante el entrenamiento. Si es "False", no se utiliza la interrupción temprana.

Valores válidos: cadena ("True" o "False").

Valor predeterminado: "False".
early_stopping_min_delta El cambio mínimo necesario para considerarse una mejora. Un cambio absoluto inferior al valor de early_stopping_min_delta no se considera mejora. Solo se usa cuando early_stopping está establecido en "True".

Valores válidos: flotante, con el rango [0.0, 1.0].

Valor predeterminado: 0.0.
early_stopping_patience

El número de epochs (fechas de inicio) para seguir entrenando sin que haya mejoras. Solo se usa cuando early_stopping está establecido en "True".

Valores válidos: número entero positivo.

Valor predeterminado: 5.
epochs

El número de fechas de inicio de capacitación.

Valores válidos: número entero positivo.

Valor predeterminado: 5 para modelos más pequeños, 1 para modelos más grandes.
epsilon

El valor épsilon para los optimizadores "adam", "rmsprop", "adadelta" y "adagrad". Se suele establecer en un valor pequeño para evitar la división por 0. No se tiene en cuenta para otros optimizadores.

Valores válidos: flotante, con el rango [0.0, 1.0].

Valor predeterminado: 1e-7.
initial_accumulator_value

El valor inicial de los acumuladores o los valores de impulso por parámetro del optimizador "adagrad". No se tiene en cuenta para otros optimizadores.

Valores válidos: flotante, con el rango [0.0, 1.0].

Valor predeterminado: 0.1.
learning_rate La tasa de aprendizaje del optimizador.

Valores válidos: flotante, con el rango [0.0, 1.0].

Valor predeterminado: 0.001.
momentum

El valor de impulso para los optimizadores "sgd" y "nesterov". No se tiene en cuenta para otros optimizadores.

Valores válidos: flotante, con el rango [0.0, 1.0].

Valor predeterminado: 0.9.
optimizer

El tipo de optimizador. Para obtener más información, consulte Optimizers en la documentación de TensorFlow.

Valores válidos: cadena, ("adam", "sgd", "nesterov", "rmsprop", "adagrad" o "adadelta").

Valor predeterminado: "adam".
reinitialize_top_layer

Si se establece en "Auto", los parámetros de la capa de clasificación superior se reinicializan durante el ajuste. Para el entrenamiento incremental, los parámetros de la capa de clasificación superior no se reinicializan, a no ser que se establezca en "True".

Valores válidos: cadena, "Auto", "True" o "False".

Valor predeterminado: "Auto".
rho

El factor de descuento para el gradiente de los optimizadores "adadelta" y "rmsprop". No se tiene en cuenta para otros optimizadores.

Valores válidos: flotante, con el rango [0.0, 1.0].

Valor predeterminado: 0.95.
train_only_on_top_layer

Si es "True", solo se refinan los parámetros de la capa de clasificación superior. Si es "False", todos los parámetros del modelo se refinan.

Valores válidos: cadena ("True" o "False").

Valor predeterminado: "False".