PaddleDetection retinanet50、101训练好的模型,预测时显存溢出
Created by: gilbert1989
-
版本、环境信息:
1)PaddlePaddle版本:v1.5
2)GPU:p4
3)系统环境:官方docker
-
复现信息:python tools/infer.py --config=configs/retinanet_r50_fpn_1x.yml --infer_dir=dataset/coco/test/
-
问题描述:batch_size=1,训练和eval没有问题,infer.py进行预测时,提示显存问题。