v0.0.44 DataParallel

Author: bartzbeielstein

notebooks/11_spot_hpt_torch.ipynb

@@ -32,7 +32,7 @@
     {
      "data": {
       "text/plain": [
-       "'11-torch_p040025_360min_10init_2023-04-26_23-24-12'"
+       "'11-torch_p040025_360min_10init_2023-04-27_00-08-17'"
       ]
      },
      "execution_count": 2,
@@ -79,7 +79,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "spotPython                                0.0.43\n",
+      "spotPython                                0.0.44\n",
       "spotRiver                                 0.0.92\n",
       "Note: you may need to restart the kernel to use updated packages.\n"
      ]
@@ -644,8 +644,8 @@
      "output_type": "stream",
      "text": [
       "Using mps device\n",
-      "[1,  2000] loss: 2.320\n",
-      "[1,  4000] loss: 1.162\n"
+      "[1,  2000] loss: 2.323\n",
+      "[1,  4000] loss: 1.159\n"
      ]
     }
    ],

pyproject.toml

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

src/spotPython/torch/net.py

@@ -21,6 +21,9 @@ def __init__(self, l1, l2, lr, batch_size, epochs):
         self.lr = lr
         self.batch_size = batch_size
         self.epochs = epochs
+        if torch.cuda.device_count() > 1:
+            print("We will use", torch.cuda.device_count(), "GPUs!")
+        self = nn.DataParallel(self)
 
     def forward(self, x):
         x = self.pool(F.relu(self.conv1(x)))