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

May I ask how long it will take to train the BP4D and DISFA data sets? I use two GeForce RTX 2080,it takes 14 days of training. #4

Open
lucky-starh opened this issue Nov 27, 2019 · 7 comments

Comments

@lucky-starh
Copy link

No description provided.

@machanic
Copy link
Owner

machanic commented Nov 27, 2019

If you use memcached and pylibmc to store the bounding box coordinates, the training will be accelarate by the cache. You must assure the cache is correctly used in the program. When I use the Nvidia Tesla P100 card, the training will be finished in 4 days at most.

@lucky-starh
Copy link
Author

Thank you. I have one more question: OutOfMemoryError. The defualt batch_size =20, can only change the bitch_size ?

@machanic
Copy link
Owner

@lucky-starh if the input image is 512 x 512, batch size =20 is too large to the GPU memory, batch size = 5 will be better.

@lucky-starh
Copy link
Author

Has been an error: Crop image error, always appear in the Img_RightCamera/SN029 or Img_RightCamera/SN029 folder

@machanic
Copy link
Owner

machanic commented Nov 27, 2019

@lucky-starh Maybe the head is not frontal position, you need to check that reason. But I remember I have already added the exception code for handling this case....

@lucky-starh
Copy link
Author

I haven't found the conversion to BGR->RGB image in get_example() function.

@machanic
Copy link
Owner

@lucky-starh I think you do not need to convert from BGR -> RGB. Have you fixed this issue?

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