v0.2.24

func_control_init accepts device

Author: bartzbeielstein

pyproject.toml

@@ -7,7 +7,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "spotPython"
-version = "0.2.23"
+version = "0.2.24"
 authors = [
   { name="T. Bartz-Beielstein", email="tbb@bartzundbartz.de" }
 ]

src/spotPython/utils/init.py

@@ -4,7 +4,7 @@
 from torch.utils.tensorboard import SummaryWriter
 
 
-def fun_control_init(task, tensorboard_path=None):
+def fun_control_init(task, tensorboard_path=None, device=None):
     """Initialize fun_control dictionary.
     Args:
         None
@@ -63,7 +63,7 @@ def fun_control_init(task, tensorboard_path=None):
         "eval": None,
         "k_folds": None,
         "optimizer": None,
-        "device": None,
+        "device": device,
         "show_batch_interval": 1_000_000,
         "path": None,
         "task": task,