0bf56c5e8db53e152e7095087499d072018283e6,src/test.py,,model_fn,#,132

Before Change

                                                   "sequence_error" : sequence_error}, 
                                                  every_n_iter=100)
    return tf.estimator.EstimatorSpec(mode=mode, loss=loss, predictions=tf.nn.softmax(logits), 
                                      train_op=None, prediction_hooks=[logging_hook])
    
def main(argv=None):

After Change

                logits,sequence_length,label,length)

        global_step = tf.convert_to_tensor(tf.train.get_or_create_global_step())
        global_step = tf.cast(global_step, tf.float32)
        sequence_error = tf.cast(sequence_error, tf.float32)

        //Get the correct format to pass tp estimator spec
        result = tf.convert_to_tensor([(tf.stack([global_step,
                                                  loss, 
                                                  label_error, 
                                                  sequence_error], axis=0))])

    return tf.estimator.EstimatorSpec(mode=mode, loss=loss, 
                                      predictions=result, 
                                      train_op=None)

Instances