Andrew's mulmodal stuff

GriffithsLab · Jan 17, 2024 · ba92903 · ba92903
1 parent bebbce0
commit ba92903
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/whobpyt/models/EEG/ParamsEEG.py b/whobpyt/models/EEG/ParamsEEG.py
@@ -9,4 +9,13 @@ def __init__(self, Lead_Field):
         #############################################
 
         self.LF = Lead_Field # This should be [num_regions, num_channels]
+
+    def to(self, device):
+        # Moves all parameters between CPU and GPU
+
+        vars_names = [a for a in dir(self) if not a.startswith('__')]
+        for var_name in vars_names:
+            var = getattr(self, var_name)
+            if (type(var) == par):
+                var.to(device)