tf 加日志

tensorflow 加日志

import logging
 
# get TF logger
log = logging.getLogger('tensorflow')
log.setLevel(logging.DEBUG)

# create formatter and add it to the handlers
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')

fh = logging.FileHandler('tensorflow.log')
fh.setLevel(logging.DEBUG)
fh.setFormatter(formatter)
log.addHandler(fh)

修改命名

import datetime

log_file = datetime.datetime.now().strftime("%Y-%m-%d-%H:%M:%S:%f:") + "create_pretraining_data_copy.log"

打印参数

  tf.logging.info("do_train:  %s", FLAGS.do_train)
  tf.logging.info("do_eval:  %s", FLAGS.do_eval)
  tf.logging.info("bert_config_file:  %s", FLAGS.bert_config_file)
  tf.logging.info("init_checkpoint:  %s", FLAGS.init_checkpoint)
  tf.logging.info("train_batch_size:  %s", FLAGS.train_batch_size)
  tf.logging.info("max_seq_length:  %s", FLAGS.max_seq_length)
  tf.logging.info("max_predictions_per_seq:  %s", FLAGS.max_predictions_per_seq)
  tf.logging.info("num_train_steps:  %s", FLAGS.num_train_steps)
  tf.logging.info("num_warmup_steps:  %s", FLAGS.num_warmup_steps)
  tf.logging.info("learning_rate:  %s", FLAGS.learning_rate)