Python中寫入訓(xùn)練日志文件并控制臺輸出的示例分析

這篇文章將為大家詳細講解有關(guān)Python中寫入訓(xùn)練日志文件并控制臺輸出的示例分析，小編覺得挺實用的，因此分享給大家做個參考，希望大家閱讀完這篇文章后可以有所收獲。

1. 背景

在深度學(xué)習(xí)的任務(wù)中，通常需要比較長時間的訓(xùn)練，因此我們會選擇離開電腦。筆者在跟蹤模型表現(xiàn)，

觀察模型accuracy 以及 loss 的時候，比較傳統(tǒng)的方法是在控制臺print輸出或者直接使用tensorboard。

但如果是你需要遠程觀察模型表現(xiàn)，那一個時刻記錄的log 文件就非常重要。

（如果你希望不在實驗室，遠程通過訪問正在訓(xùn)練網(wǎng)絡(luò)的服務(wù)器的Jupyter Notebook ，實時查看實驗進度。請參考——遠程連接服務(wù)器端Jupyter Notebook）

2. logging 函數(shù)

創(chuàng)建 .log 文件并將一些數(shù)據(jù)在控制臺輸出可以通過logging 函數(shù)實現(xiàn)。

廢話不多，直接貼代碼：一下函數(shù)可以在你想要保存日志文件的路徑——output_dir下生成一個.log 文件，文件名以年.月.日.時.分命名，以免重復(fù)。

import logging
import time
import os
def log_creater(output_dir):
  if not os.path.exists(output_dir):
    os.makedirs(output_dir)
  log_name = '{}.log'.format(time.strftime('%Y-%m-%d-%H-%M'))
  final_log_file = os.path.join(output_dir,log_name)
  # creat a log
  log = logging.getLogger('train_log')
  log.setLevel(logging.DEBUG)

  # FileHandler
  file = logging.FileHandler(final_log_file)
  file.setLevel(logging.DEBUG)

  # StreamHandler
  stream = logging.StreamHandler()
  stream.setLevel(logging.DEBUG)

  # Formatter
  formatter = logging.Formatter(
    '[%(asctime)s][line: %(lineno)d] ==> %(message)s')

  # setFormatter
  file.setFormatter(formatter)
  stream.setFormatter(formatter)

  # addHandler
  log.addHandler(file)
  log.addHandler(stream)

  log.info('creating {}'.format(final_log_file))
  return log

3. 結(jié)果

保存后的日志文件差不多就是這樣：

關(guān)于“Python中寫入訓(xùn)練日志文件并控制臺輸出的示例分析”這篇文章就分享到這里了，希望以上內(nèi)容可以對大家有一定的幫助，使各位可以學(xué)到更多知識，如果覺得文章不錯，請把它分享出去讓更多的人看到。