Skip to content

[REQUEST] Custom Callbacks #865

@Zeyi-Lin

Description

@Zeyi-Lin

🤩 Features description [Please make everyone to understand it]

0.5.0中,我们将实现一个新特性 —— callbacks,大致API交互如下:

import swanlab

swanlab.init(callbacks=[...])

callbacks支持传入一系列自定义类,这些类需要继承自SwanKitCallback类(swankit.callback.SwanKitCallback)。SwanKitCallback支持一系列生命周期函数,以便在SwanLab执行如initlogfinish等操作时,插入自定义行为。

其中,我们需要在swanlab/plugin/notifaction.py下维护一些官方提供的Callback(我们更愿意定义为插件),以更好地让开发者快速集成。

第一批我们希望的Callbacks是通知类,包括:

  • EmailCallback:训练完成或异常时,发送通知邮件
  • LarkCallback:训练完成或异常时,发送给飞书机器人进行通知
  • DingCallback:训练完成或异常时,发送给钉钉进行通知
  • WXWorkCallback:训练完成或异常时,发送给企业微信进行通知

记录类:

  • CSVWriterCallback:记录实验信息到本地csv文件中

Metadata

Metadata

Assignees

Labels

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions