MXNet (0) 聊聊 MXNet
個人覺得 MXNet 基於 PyTorch 做了不少改進,進而發展出一套自己的生態系。
目前發現 MXNet 有幾個特點:
- 自動判斷 input 的 channel size
- 可以輕鬆寫單機多 GPU 的程式
- 以 CV 而言備齊了各種基礎的資料 loader (eg. object detection, pose estimation)
- 提供算圖 compile 及優化的機制
目前 MXNet 雖然還不支援 Conda ,但其優勢就是涵蓋了 CV 的基礎 dataset 和 models 寫起來簡便又不失運行速度。
接下來會慢慢整理 MXNet 的相關文章 ^_^
留言
張貼留言