使用pre_trained model进行微调

Shuyu-XJTU / APTM

The official code of "Towards Unified Text-based Person Retrieval: A Large-scale Multi-Attribute and Language Search Benchmark"

https://arxiv.org/abs/2306.02898

MIT License

139 stars 12 forks source link

Closed laiping-lp closed 5 months ago

laiping-lp commented 6 months ago

请问为什么使用pre_trained model在cuhk-pedes这些数据集进行微调时，一直会报out of memory的错误，我的设备如下 QQ截图20240509112205

littlexinyi commented 5 months ago

估计是显存不够

Shuyu-XJTU commented 5 months ago

可以试着调小batch size