MegEngine / Models

采用MegEngine实现的各种主流深度学习模型
Other
303 stars 99 forks source link

COCO数据集测试速度异常 #105

Open er-muyue opened 3 years ago

er-muyue commented 3 years ago

背景

COCO数据集测试速度异常

image

任务描述

按照首页readme安装megengine和Models,从 hub.py 下载 预训练权重,直接在coco数据集上进行测试。代码未做任何改动,采用8卡测试,最终结果mAP为42.6(正常),但是耗时将近6m30s(不正常),正常8卡时间应该在1min左右

python:3.6 megengine版本:v1.4.0

目标

希望多卡测试速度正常,请确认问题并帮忙修复

JsBlueCat commented 3 years ago

请帮忙看下nvidia-smi 的占用情况

er-muyue commented 3 years ago

看起来是正常的,但是GPU利用率较低 image