import tensorflow as tf

            eval_input_fn = None
        '''
        # Add hooks
        train_hooks = [
            tf.train.StopAtStepHook(last_step=params.train_steps),
            tf.train.NanTensorHook(loss),  # Monitors the loss tensor and stops training if loss is NaN
            tf.train.LoggingTensorHook(
                {
                    "step": global_step,
                    "loss": loss,
