search

PaddlePaddle / models

Officially maintained, supported by PaddlePaddle, including CV, NLP, Speech, Rec, TS, big models and so on.
Apache License 2.0
6.92k stars 2.91k forks source link

window上metric_learning模型进行eval报MemoryError #2504

Open JiaXiao243 opened 5 years ago

JiaXiao243 commented 5 years ago

在window机器上，paddle==1.5，在\PaddleCV\metric_learning，执行python eval.py --model=ResNet50 --batch_size=16 --pretrained_model=output，对模型进行eval，报错信息如下： image

wopeizl commented 5 years ago

recall_topk内存占用过大，导致内存不足，需要优化此函数

jiayifang1 commented 5 years ago

你好，我在训练过程中也出现了”MemoryError“的问题，请问你的这个问题解决了吗？具体怎样优化函数呢