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

Problem about multi processing #5

Open
SilencerChen opened this issue Mar 24, 2018 · 1 comment
Open

Problem about multi processing #5

SilencerChen opened this issue Mar 24, 2018 · 1 comment

Comments

@SilencerChen
Copy link

Thanks for your great work at first.
I run you code and find out the usage rate of my GPU is always 0% when calculate the scores of the val captions. I train to use multiprocessing and need to add "num_works" in torch.utils.data.DataLoader but I find you write the class all by yourself.
So is there a way to use the mutiprocessing?

@plsang
Copy link
Collaborator

plsang commented Mar 26, 2018

  • I think the COCO Caption Evaluation code does not support running on GPU.
  • You can implement a DataLoader class to support parallel loading; however, I don't think it improves the speed significantly since data loading is quite fast.
  • You may need to run on multiple GPUs; however, I haven't tried to implement this. Any contribution is welcome.

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