wandb

import wandb
import time

启动/结束

wandb.init(project='HPA', name=time.strftime('%m%d%H%M%S'))

wandb.finsh()

日志记录

#log_train = {}
#log_train['epoch'] = 0
#log_train['batch'] = 0
#log_train['acc'] = ***
wandb.log(log_train)

监控

• model可以是单个，也可以是多个，包含在list中
• log为记录的指标，可以是 “gradients”、”parameters”、”all” 或 None， 默认值为 “gradients”
  • 通过监控gradients、parameters观察模型是否趋于稳定

wandb.watch(model, log="gradients", log_freq=1000, log_graph=False)