UniVL icon indicating copy to clipboard operation
UniVL copied to clipboard

Non-Configurable GPU Count via Arguments

Open willyfh opened this issue 1 year ago • 0 comments

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.

willyfh avatar Feb 14 '24 16:02 willyfh