fix typo in example code

lanl · Apr 5, 2024 · e84f93c · e84f93c
1 parent 3ddc0c6
commit e84f93c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/feature_extraction.py b/examples/feature_extraction.py
@@ -8,7 +8,7 @@
 import torch
 
 if torch.cuda.is_available():
-    device = torch.cuda.get_default_device
+    device = torch.device(0)
 else:
     device = torch.device("cpu")
 import hippynn