GPU usage

[SzhangS]

您好， 我在预处理阶段使用SMPLer-X l32, gpu占用峰值能达到55G，请问这合理吗？或者您有什么建议？

[theFoxofSky]

请问预处理阶段是指Inference还是Training还是SMPL提取？

[SzhangS]

Inference阶段

[SzhangS]

训练和推理阶段的数据预处理 ，应该是一致的吧？ 准确描述的话，我是在构建推理数据时候，按照prepare_pose描述的那样数据被smpler-x处理的时候，GPU峰值达到55G

[theFoxofSky]

训练和推理阶段的数据预处理 ，应该是一致的吧？ 准确描述的话，我是在构建推理数据时候，按照prepare_pose描述

Prepare pose这个是预先在集群上跑好的，具体要多少显存我也不清楚，@Wangbenzhi 你来看看

[Wangbenzhi]

不是很正常，V100就可以完成数据推理， 请问你的分辨率是多少？

[SzhangS]

输入mmdet的帧分辨率时（772， 1200）

[Wangbenzhi]

我重新测试了显存使用，mmdet默认分辨率是img_scale=(1333, 800), 这个分辨率下显存开销～30G。 具体参照prepare_pose/smplerX/common/utils/human_model_files/mmdet_faster_rcnn_r50_fpn_coco.py。 如有需要，可以降低img_scale， 不过55G肯定是不正常的，可以debug看下哪一步显存开销较大。