feature: Added new device

Flippchen · Dec 30, 2023 · 9c97ef2 · 9c97ef2
1 parent f3b3790
commit 9c97ef2
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 4 deletions.
diff --git a/training/old/vision.py b/training/old/vision.py
@@ -115,7 +115,9 @@
 
 # Train model
 epochs = 20
-with tf.device('/GPU:0'):
+device = tf.test.gpu_device_name() if tf.test.is_gpu_available() else '/CPU:0'
+print("Using Device:", device)
+with tf.device(device):
  history = model.fit(
  train_ds,
  validation_data=val_ds,

diff --git a/training/old/with_augmentation.py b/training/old/with_augmentation.py
@@ -69,7 +69,9 @@
 
 # Train model
 epochs = 20
-with tf.device('/GPU:1'):
+device = tf.test.gpu_device_name() if tf.test.is_gpu_available() else '/CPU:0'
+print("Using Device:", device)
+with tf.device(device):
  history = model.fit(
  train_ds,
  validation_data=val_ds,

diff --git a/training/old/without_augmentation.py b/training/old/without_augmentation.py
@@ -62,7 +62,9 @@
 
 # Train model
 epochs = 20
-with tf.device('/GPU:1'):
+device = tf.test.gpu_device_name() if tf.test.is_gpu_available() else '/CPU:0'
+print("Using Device:", device)
+with tf.device(device):
  history = model.fit(
  train_ds,
  validation_data=val_ds,

diff --git a/training/pre_filter.py b/training/pre_filter.py
@@ -115,7 +115,9 @@
 
 # Train model
 epochs = 15
-with tf.device('/GPU:0'):
+device = tf.test.gpu_device_name() if tf.test.is_gpu_available() else '/CPU:0'
+print("Using Device:", device)
+with tf.device(device):
  history = model.fit(
  train_ds,
  validation_data=val_ds,