Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

OOM：out of memory #2

Closed
ray0809 opened this issue Jun 20, 2019 · 2 comments
Closed

OOM：out of memory #2

ray0809 opened this issue Jun 20, 2019 · 2 comments

Comments

@ray0809
Copy link

ray0809 commented Jun 20, 2019

您好，当我取cifar10整体训练的5w张拿来训练，存在一个显存溢出问题，请问你这边训练是用了多少数据
@ray0809 ray0809 changed the title OMM：out of memory OOM：out of memory Jun 20, 2019
@TreezzZ
Copy link
Owner

TreezzZ commented Jun 20, 2019

您好，当我取cifar10整体训练的5w张拿来训练，存在一个显存溢出问题，请问你这边训练是用了多少数据

按照论文原文，取1000张作为查询图像，5000张作为训练图像。训练完成后，对所有图像生成hash code，然后用1000个查询图像计算mAP。
我记得我最开始的时候也是直接用cifar10所有的数据来训练的，没有溢出的问题。你检查一下是不是batch-size设置的太大了？

@ray0809
Copy link
Author

ray0809 commented Jun 21, 2019

发现是我自己改动了代码的原因，感谢你的回答～

@ray0809 ray0809 closed this as completed Jun 21, 2019
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@TreezzZ @ray0809