MegEngine 天元 深度学习框架发布

司承业
2023-12-01

国内即百度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

后续补充

 类似资料: