预先知识

${CONFIG_FILE}：config/里面的文件

config/faster_rcnn_r50_fpn_1x_coco.py

${CHECKPOINT_FILE}：模型权重所在位置

checkpoints/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth

[–out ${RESULT_FILE}]：测试生成的文件输出位置

[–eval ${EVAL_METRICS}]：选用的测试方法

${GPU_NUM}：GPU数量

测试数据集

# single-gpu
python tools/test.py ${CONFIG_FILE} ${CHECKPOINT_FILE} [--out ${RESULT_FILE}] [--eval ${EVAL_METRICS}] [--show]# multi-gpu testing
./tools/dist_test.sh ${CONFIG_FILE} ${CHECKPOINT_FILE} ${GPU_NUM} [--out ${RESULT_FILE}] [--eval ${EVAL_METRICS}]

模型训练

单机单GPU训练

python tools/train.py ${CONFIG_FILE}

举例：

python tools/train.py ./configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py

如需指定工作目录，后接参数：--work_dir${WORK_DIR}

单机多GPU训练

./tools/dist_train.sh ${CONFIG_FILE} ${GPU_NUM} [optional arguments]

举例：

./tools/dist_train.sh ./configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py 4

可选参数：
--validate：训练过程中，每隔k代执行一次评估（默认为1）
--work_dir ${WOR_DIR}：指定工作目录
--resume_from ${CHECKPOINT_FILE}：从先前的检查点文件恢复

多机多GPU训练

使用slurm集群管理：

./tools/slurm_train.sh ${PARTITION} ${JOB_NAME} ${CONFIG_FILE} ${WORK_DIR} [${GPUS}]

举例：16GPU，test分区，训练faster R-CNN

./tools/slurm_train.sh test Faster_r50_1x configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py /home/xxx/faster_rcnn_r50_fpn_1x 16

Reference

MMDetection中文文档——2.入门

MMDetection——GPU训练相关推荐

在服务器上利用mmdetection来训练自己的voc数据集
在服务器上利用mmdetection来训练自己的voc数据集服务器上配置mmdetection环境在服务器上用anaconda配置自己的环境进入自己的虚拟环境,开始配置mmdetection 跑 ...
【MMDetection】——训练个人数据集
文章目录 1.数据集格式及存放 2.修改两处 3.用训练命令生成配置文件 4.正式训练开始 5.报错记录 6.模型评价测试(VOC指标mAP.COCO指标AP) 7.绘制每个类别bbox 的结果曲线图 ...
pytorch 多GPU训练总结（DataParallel的使用）
版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明. 本文链接:https://blog.csdn.net/weixin_40087578/arti ...
Pytorch中多GPU训练指北
前言在数据越来越多的时代,随着模型规模参数的增多,以及数据量的不断提升,使用多GPU去训练是不可避免的事情.Pytorch在0.4.0及以后的版本中已经提供了多GPU训练的方式,本文简单讲解下使用P ...
MinkowskiEngine多GPU训练
MinkowskiEngine多GPU训练目前,MinkowskiEngine通过数据并行化支持Multi-GPU训练.在数据并行化中,有一组微型批处理,这些微型批处理将被送到到网络的一组副本中. ...
从800个GPU训练几十天到单个GPU几小时，看神经架构搜索如何进化
点击上方"视学算法",选择加"星标"或"置顶" 重磅干货,第一时间送达本文转载自:机器之心作者:Erik Lybecker | 参 ...
多gpu训练梯度如何计算，求和是否要求平均
作者:智星云服务链接:https://www.zhihu.com/question/271226455/answer/1521784627 来源:知乎著作权归作者所有.商业转载请联系作者获得授权, ...
47分钟，BERT训练又破全新纪录！英伟达512个GPU训练83亿参数GPT-2 8B
关注上方"深度学习技术前沿",选择"星标公众号", 资源干货,第一时间送达! 具有92个DGX-2H节点的NVIDIA DGX SuperPOD通过在短短47分 ...
torch distributed 多GPU训练笔记
DistributedDataParallel (既可单机多卡又可多机多卡) 先奉上官网nn.DistributedDataParallel(model)链接 https://pytorch.org/ ...

MMDetection——GPU训练