PaddleX
PaddleX copied to clipboard
Predictor类初始化参数新增gpu_mem: GPU显存大小
Predictor类初始化参数新增gpu_mem: GPU显存大小
Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.
You have signed the CLA already but the status is still pending? Let us recheck it.
HI @angel-yi 函数参数新增时,从最后面开始追加,避免对已有调用此函数的代码带来影响
好的,谢谢,是我大意了
另外在默认值的选取上,为什么是设定的200呢
if use_gpu:
# 设置GPU初始显存(单位M)和Device ID
config.enable_use_gpu(200, gpu_id)