From 57d1fe56c560367a215f495dbc0cf7c634588bc2 Mon Sep 17 00:00:00 2001 From: Artem Agrkov <99346774+agarkovv@users.noreply.github.com> Date: Thu, 24 Oct 2024 12:38:13 +0300 Subject: [PATCH] Typo fix in train_pytorch.py (#1758) --- references/detection/train_pytorch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/references/detection/train_pytorch.py b/references/detection/train_pytorch.py index 091d257898..8682ddb60f 100644 --- 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()