Knock Knock！你的模型已訓練完成……

選自Github

作者：huggingface

機器之心編譯

參與：淑婷、路雪

訓練出了問題你卻不知道？訓練結束了你還在傻傻等待？這裡有一個好用的工具，兩行代碼即可實時通知模型訓練狀況。本文介紹了一個小型的代碼庫 Knock Knock，當你的模型訓練完成或者訓練過程出現問題時，它會及時通知你。而你只需要寫兩行代碼。

項目地址：https://github.com/huggingface/knockknock

在訓練深度學習模型時，我們通常會使用早停法。除了粗略的估計，你很難預測出訓練什麼時候會結束。因此，為模型訓練設置自動通知就很有意思了。而且當訓練因為未知原因而中途崩潰時，收到通知就更重要了。

安裝

用 pip 或類似的工具安裝。

該代碼僅用 Python3.6 測試過。

使用

該庫可無縫使用，只需對代碼做最小的修改：你只需在主函數調用上加一個裝飾器。

現在有兩種設置通知的方式：郵件和 Slack。

郵件

郵件服務要依賴 Yagmail，這是一個 GMAIL/SMTP 客戶端。你需要一個gmail郵件地址來使用它。最好創建一個新的郵件地址（不要使用常用地址），因為你需要修改賬戶的安全設置，以允許該 Python 庫打開不太安全的 APP（選中 Allow less secure apps）進而訪問它。

如果未指定 sender_email，則 recipient_email 也可以用於發送郵件。

注意，啟用此功能會向你詢問發件人的郵箱密碼。密碼將通過 keyring Python 庫被安全地存儲在系統 keyring 服務中。

Slack

你還可以使用 slack 來獲取通知。你必須提交 Slack 房間的 webhook URL 和用戶 id（用戶 id 為可選項，如果你想添加自己或其他人則選擇該項）。

你還可以指定一個可選參數來添加特定的人：user_mentions=[, ]。

本文為機器之心編譯，轉載請聯繫本公眾號獲得授權。

------------------------------------------------

喜歡這篇文章嗎？立刻分享出去讓更多人知道吧！