Typo fix in train_pytorch.py (#1758)

mindee · Oct 24, 2024 · 57d1fe5 · 57d1fe5
1 parent 251aa22
commit 57d1fe5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/references/detection/train_pytorch.py b/references/detection/train_pytorch.py
@@ -238,7 +238,7 @@ def main(args):
     elif torch.cuda.is_available():
         args.device = 0
     else:
-        logging.warning("No accessible GPU, targe device set to CPU.")
+        logging.warning("No accessible GPU, target device set to CPU.")
     if torch.cuda.is_available():
         torch.cuda.set_device(args.device)
         model = model.cuda()