DeepLog

极简的深度学习日志记录工具

安装

下载源码

开源地址：https://github.com/MacroHongZ/DeepLog

本地安装

# 需先进入源码文件夹
pip install -e .

查看安装

pip list
# 可以看到包名 deeplog

主要功能

from deeplog import DeepLog, Config

# 初始化
Log = DeepLog()
config = Config()

# 配置模型超参数
config.lr = 0.01
config.bathch_size = 50
config.epoch = 100

# 打印模型超参数
config.print_parameters()
'''
lr :  0.01
bathch_size :  50
epoch :  100
'''

# lode args from argparse
config .from_argparse(args)

# 记录日志,记录 loss 和 metrics 等重要信息
for i in range(50):    
    Log.log("loss", i)
    Log.log("Acc", (i*0.5)**2)
    
# 基于 Elegant-Plot（https://github.com/MacroHongZ/Elegant-Plot） 的可视化
Log.visualization(item="loss")
'''
item: 默认值 all，字符串，会将记录的所有信息可视化。也可以指定可视化项目。
Log.get_log_keys() 可以查看所有记录的项目
'''

# 保存日志
Log.save(config=config, config_save=True)
'''
config: 默认值 None，需传入 Config 对象，传入后会在日志中记录超参数。
config_save：默认值 False，布尔对象。值为 True 会单独存储超参数，用于之后加载复现结果。
'''

# 从文件加载超参数，用于复现结果
config = Log.load_config("2022-09-02_13-11-17config.pickle")
config.print_parameters()

# 重新解析log文件，用于绘图
Log.load_logs('MyProject_log//2022-09-13_14-54-56log.txt')
Log.visualization(item="loss")

日志展示

绘图展示