Fixup

Logiquo · Logiquo · commit 6ad04588f29e · 2025-12-25T07:13:29.000-05:00
diff --git a/examples/memtest.py b/examples/memtest.py
@@ -67,7 +67,7 @@
 
         trainer = Trainer(
             model=model,
-            device="cuda:0",  # or "cpu"
+            device="cpu",  # or "cpu"
             metrics=["pr_auc", "roc_auc", "accuracy", "f1"],
         )
 

Original file line number	Diff line number	Diff line change
`@@ -67,7 +67,7 @@`
`67`	`67`
`68`	`68`	`trainer = Trainer(`
`69`	`69`	`model=model,`
`70`		`- device="cuda:0", # or "cpu"`
	`70`	`+ device="cpu", # or "cpu"`
`71`	`71`	`metrics=["pr_auc", "roc_auc", "accuracy", "f1"],`
`72`	`72`	`)`
`73`	`73`