search

IrisRainbowNeko / HCP-Diffusion

A universal Stable-Diffusion toolbox
Apache License 2.0
896 stars 75 forks source link

希望能接入wandb的训练日志 #14

Closed petercham closed 1 year ago

petercham commented 1 year ago

https://wandb.ai/ 在配置文件里面填写上appkey，配置测试图的Prompt文件路径，自动每隔多少步出一张测试图

IrisRainbowNeko commented 1 year ago

wandb login之后，在logger模块中添加 hcpdiff.loggers.WanDBLogger 即可

logger:
  -
    _target_: hcpdiff.loggers.CLILogger
    _partial_: True
    out_path: 'train.log'
    log_step: 20
  - _target_: hcpdiff.loggers.WanDBLogger
    _partial_: True
    log_step: 5