MegEngine 天元 深度學習框架發佈

國內即百度paddlepaddle 發佈之後,曠視科技又發佈了MagEngine 深度學習框架。目前僅僅支持ubuntu 系統。python 3.5 以上。cuda 10.0。目前代碼已在github上開源,地址是https://github.com/MegEngine

 

目前試驗了一下物體檢測的代碼。在服務器上運行發現,多卡運行有點問題。

具體運行代碼。

python3 tools/train.py -f retinanet_res50_1x_800size.py   -n 10 --batch_size 2 --dataset-dir

後續補充

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章