MXNet (0) 聊聊 MXNet

個人覺得 MXNet 基於 PyTorch 做了不少改進,進而發展出一套自己的生態系。

目前發現 MXNet 有幾個特點:

  1. 自動判斷 input 的 channel size
  2. 可以輕鬆寫單機多 GPU 的程式
  3. 以 CV 而言備齊了各種基礎的資料 loader (eg. object detection, pose estimation)
  4. 提供算圖 compile 及優化的機制

目前 MXNet 雖然還不支援 Conda ,但其優勢就是涵蓋了 CV 的基礎 dataset 和 models 寫起來簡便又不失運行速度。

接下來會慢慢整理 MXNet 的相關文章 ^_^

留言

熱門文章