國內即百度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
後續補充