[references] fix typo (#1492)

mindee · Mar 1, 2024 · 707ca42 · 707ca42
1 parent aae1087
commit 707ca42
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/references/recognition/train_pytorch.py b/references/recognition/train_pytorch.py
@@ -248,7 +248,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()