Bring TensorBoard to MXNet

個人認為，參與開源項目最好的方式是從使用它開始。由於在團隊里「特立獨行」地用 MXNet 作為一個項目中的深度學習框架，遇到問題都是自己想辦法解決或者通過社區的夥伴幫忙（大部分問題在 issue 上都能找到，提問也能很快得到反饋），慢慢對這個工具熟悉起來。

因為工作上需要一些可視化來幫助自己了解模型的情況，而且發現 TensorBoard 還挺好用，因此便有了為 MXNet 也開發一款可視化工具的想法。幾經搜尋之後發現有不少使用者也有這樣的需求，有些同學想重新搞一套，也有同學提出能否把 TensorBoard 借用過來；我喜歡這個想法，因此著手開始研究是否可行，也在 issue 上發起了一個討論 Building a visualization tool for MXNet · Issue #4003 · dmlc/mxnet 主要從 motivation 和幾個關鍵的問題出發，向大家徵求意見。目標是找到一種易維護的方案，讓用戶不需整個編譯/安裝 TensorFlow 就可以試用 TensorBoard 的 logging 和 rendering 功能；提供 Python 介面，也可以用於一般的可視化任務。

通過一段時間的努力，我們完成了主要的幾個目標（下一步的目標是支持全功能，並移植入 MXNet 並提供簡易的安裝方案），請大家一起來試用 https://github.com/dmlc/tensorboard

同時我們寫了一篇 tutorial 來演示 TensorBoard 的使用，主題是通過可視化對深度學習中的 vanishing grandient 問題有一個直觀的認識： Understanding the vanishing gradient problem through visualization

對其中細節感興趣的同學可以移步 DMLC 的博客看我們最新的 Post：

Bring TensorBoard to MXNet

最後，謝謝 DMLC 的小夥伴~ 尤其是 Eric @解浚源 的幫助！

推薦閱讀：