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

Multi-CPU parallel training #4

Open
Taylor-Liu opened this issue Jul 16, 2020 · 1 comment
Open

Multi-CPU parallel training #4

Taylor-Liu opened this issue Jul 16, 2020 · 1 comment

Comments

@Taylor-Liu
Copy link

Hi, Wenlong. Thanks for sharing your code.

When I reproduced your code, I found that only one CPU was used for training. The training speed was a bit slow. Can your code train with multiple CPUs in parallel? I can't find the corresponding option in your code configuration.

@huangwl18
Copy link
Owner

Thanks for the interest in the paper! The code is not configured for multi-CPU training. Only vectorized environments (from OpenAI baselines) and torchfold (dynamic batching) are currently used for speedup in training.

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