三行代码开启模型训练通知

简介

如果你经常在大量数据上训练模型的话，那么这样一个工具肯定很合你的胃口。这个项目叫做knockknock，它的功能只有一个：通知你训练结束了，并且附带训练的结果。

当前支持邮件、短信、微信群、钉钉群等通知方式，只需要三行代码，就能实现功能。

效果展示

为了让你有看下去的动力，先展示一下最终的成果(钉钉)：

knockknock在钉钉群中的配置方式

之所以选择使用钉钉，是因为它的通知声音比较好听🤭。配置过程：

1. 建立钉钉群（最好是在电脑）

2. 添加机器人

   2.1 群设置$\rightarrow $智能群助手$\rightarrow $添加机器人

   2.2 添加自定义机器人

   2.3 自定义名字和头像

   2.4 选择一种加密方式，推荐选择加签

   2.5 记录好机器人的url和加签生成的密钥

3. 在notebook或者python虚拟环境中安装knockknock

   !pip install knockknock(notebook)
   or
   pip install knockknock(虚拟环境)

4. 导入dingtalk_sender

   from knockknock import dingtalk_sender

5. 在要跑的类上添加以下代码

   webhook_url = "https://oapi.dingtalk.com/robot/send?access_token=..."
   @dingtalk_sender(webhook_url=webhook_url, secret="加签生成的密钥", keywords=["随便填"])
   def train_your_nicest_model(your_nicest_parameters):
       import time
       time.sleep(10)
       return {'loss': 0.9} # Optional return value

6. 运行train_your_nicest_model，得到结果

   

参考文献

本文主要参考自官方github仓库

以及阿里钉钉开发者平台