fixed: check for CPU

Löffler, Hannes · Löffler, Hannes · commit 31b90474cabb · 2025-11-20T09:24:13.000+01:00
diff --git a/reinvent/runmodes/RL/run_staged_learning.py b/reinvent/runmodes/RL/run_staged_learning.py
@@ -186,7 +186,7 @@ def run_staged_learning(
                 intrinsic_penalty=intrinsic_penalty,
             )
 
-            if hasattr(torch, device.type):
+            if hasattr(torch, device.type) and device.type != "cpu":
                 gpu = getattr(torch, device.type)
                 free_memory, total_memory = gpu.mem_get_info()
                 free_memory //= 1024**2

Original file line number	Diff line number	Diff line change
`@@ -186,7 +186,7 @@ def run_staged_learning(`
`186`	`186`	`intrinsic_penalty=intrinsic_penalty,`
`187`	`187`	`)`
`188`	`188`
`189`		`- if hasattr(torch, device.type):`
	`189`	`+ if hasattr(torch, device.type) and device.type != "cpu":`
`190`	`190`	`gpu = getattr(torch, device.type)`
`191`	`191`	`free_memory, total_memory = gpu.mem_get_info()`
`192`	`192`	`free_memory //= 1024**2`