La función de decadencia de la tasa de aprendizajetf.train.exponential_decay toma unadecay_steps parámetro. Para disminuir la tasa de aprendizaje cadanum_epochs, usted estableceríadecay_steps = num_epochs * num_train_examples / batch_size. Sin embargo, al leer datos de.tfrecords archivos, no sabes cuántos ejemplos de entrenamiento hay dentro de ellos.

Llegarnum_train_examples, tú podrías:

Configurar untf.string_input_producer connum_epochs=1.Ejecute esto a través detf.TFRecordReader/tf.parse_single_example.Haga un bucle y cuente cuántas veces produce algo de salida antes de detenerse.

Sin embargo, esto no es muy elegante.

¿Existe una manera más fácil de obtener el número de ejemplos de capacitación de un.tfrecords presentar o establecer la disminución de la tasa de aprendizaje en función de las épocas en lugar de los pasos?