PaddlePaddle / PaddleClas

A treasure chest for visual classification and recognition powered by PaddlePaddle
Apache License 2.0
5.42k stars 1.16k forks source link

Specify GPU memory for paddleclas.PaddleClas #2913

Open glitchyordis opened 1 year ago

glitchyordis commented 1 year ago

欢迎您使用PaddleClas并反馈相关问题,非常感谢您对PaddleClas的贡献! 提出issue时,辛苦您提供以下信息,方便我们快速定位问题并及时有效地解决您的问题: PaddleClas版本以及PaddlePaddle版本:paddleclas==2.5.1 , paddlepaddle-gpu==2.4.2.post116 涉及的其他产品使用的版本号:Not applicable. Only imported paddleclas 训练环境信息: a. 具体操作系统,On Windows 10 but not training. 4GB GPU b. Python版本号,python 3.9 c. CUDA/cuDNN版本,CUDA 11.6 完整的代码(相比于repo中代码,有改动的地方)、详细的错误信息及相关log

Is there a way to specify gpu_mem like CLI when using model = paddleclas.PaddleClas(model_name="text_image_orientation")?

cuicheng01 commented 11 months ago

报错的命令是?