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.

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

Question about machine #18

Closed
SummerTrains opened this issue Nov 11, 2016 · 3 comments
Closed

Question about machine #18

SummerTrains opened this issue Nov 11, 2016 · 3 comments

Comments

@SummerTrains
Copy link

First of all, thank you for your job!
However, when I run script_rpn_pedestrian_VGG16_caltech.m , my computer jamed and can not do anything, staying in the stage "stage one RPN"! Then, I waited for about 1 hour, it's still jamed.
So I just want to ask the minimum requirement about machine.
Here is my machine : One Titan x GPU; 32G memory. Is enough for this experiment?
Thank you very much!

@zhangliliang
Copy link
Owner

Hi,

Actually the problem might lie on the CPU memory (not the GPU memory).

The RPN need to pre-compute and save all the anchors for all images, and store them in the memory during the training time.

The Caltech 10x dataset contains 4w+ images, and it might need a large memory.

Thus, it might help when reduce the training data, e.g. reduce to Caltech 3x or Caltech 5x.

@SummerTrains
Copy link
Author

@zhangliliang Thank you!
I'm reading your code.
So which configuration can change Caltech 10x to Caltech 5x or 3x ? Can you tell me now ?
Thank you!

@zhangliliang
Copy link
Owner

Hi,

Change the skip in the extract_img_anno.m.

Skip = 3 coresponds to caltech10x, skip = 6 corespond a to caltech5x.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants