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

Non-Configurable GPU Count via Arguments #47

Open
willyfh opened this issue Feb 14, 2024 · 0 comments
Open

Non-Configurable GPU Count via Arguments #47

willyfh opened this issue Feb 14, 2024 · 0 comments

Comments

@willyfh
Copy link

willyfh commented Feb 14, 2024

In this code, https://github.com/microsoft/UniVL/blob/main/main_task_caption.py#L143-L145 , the value of n_gpu is determined by the actual number of GPUs present in the system. However, there are scenarios where utilizing only a single GPU, despite having multiple GPUs available, could be preferable. Rather than forcibly adjusting the args.n_gpu based solely on the detected number of GPUs, it could be beneficial to leave this decision to the user.

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

1 participant