国内即百度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
后续补充