-
Notifications
You must be signed in to change notification settings - Fork 152
Labels
💪 enhancementNew feature or requestNew feature or request
Description
🤩 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执行如init
、log
、finish
等操作时,插入自定义行为。
其中,我们需要在swanlab/plugin/notifaction.py
下维护一些官方提供的Callback(我们更愿意定义为插件),以更好地让开发者快速集成。
第一批我们希望的Callbacks是通知类,包括:
- EmailCallback:训练完成或异常时,发送通知邮件
- LarkCallback:训练完成或异常时,发送给飞书机器人进行通知
- DingCallback:训练完成或异常时,发送给钉钉进行通知
- WXWorkCallback:训练完成或异常时,发送给企业微信进行通知
记录类:
- CSVWriterCallback:记录实验信息到本地csv文件中
KMnO4-zxKMnO4-zxKMnO4-zxKMnO4-zx
Metadata
Metadata
Assignees
Labels
💪 enhancementNew feature or requestNew feature or request