From 039d7be9986c5d2e9ba74b1a1c1edd2696de6789 Mon Sep 17 00:00:00 2001 From: Erick Matsen Date: Wed, 13 Nov 2024 11:36:44 -0800 Subject: [PATCH] make format --- netam/common.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/netam/common.py b/netam/common.py index db61ec3c..ca6fc501 100644 --- a/netam/common.py +++ b/netam/common.py @@ -204,9 +204,9 @@ def pick_device(gpu_preference=None): If gpu_preference is a string, use the device with that name. This is considered a strong preference from a user who knows what they are doing. - If gpu_preference is an integer, this is a weak preference for a numbered - GPU. If CUDA is available, use the least used GPU, and if all are idle use - the gpu_index modulo the number of GPUs. If gpu_index is None, then use a random GPU. + If gpu_preference is an integer, this is a weak preference for a numbered GPU. If + CUDA is available, use the least used GPU, and if all are idle use the gpu_index + modulo the number of GPUs. If gpu_index is None, then use a random GPU. """ # Strong preference for a specific device.