From f0855d92f95e8cc9d8903bbeb79cf8c1fed7d3b5 Mon Sep 17 00:00:00 2001
From: Paul-Walcher <149962897+Paul-Walcher@users.noreply.github.com>
Date: Wed, 10 Jul 2024 07:51:40 +0200
Subject: [PATCH] Update trainer.py

In trainer.py renamed "device_type"-variable inside comparison to "device".
---
 trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/trainer.py b/trainer.py
index 91131a9..bed2621 100644
--- a/trainer.py
+++ b/trainer.py
@@ -148,7 +148,7 @@ def _set_device(args):
     gpus = []
 
     for device in device_type:
-        if device_type == -1:
+        if device == -1:
             device = torch.device("cpu")
         else:
             device = torch.device("cuda:{}".format(device))