From 82a04e31d5c60cbbc602e2163fd4499f64cbdef5 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 6 Oct 2021 12:57:03 -0500
Subject: [PATCH] minor
---
modules/deeplearning/icing_cnn.py | 22 +++++++++++-----------
1 file changed, 11 insertions(+), 11 deletions(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 1b5c34b8..3b582a58 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -217,17 +217,17 @@ class IcingIntensityNN:
tf.debugging.set_log_device_placement(LOG_DEVICE_PLACEMENT)
- gpus = tf.config.experimental.list_physical_devices('GPU')
- if gpus:
- try:
- # Currently, memory growth needs to be the same across GPUs
- for gpu in gpus:
- tf.config.experimental.set_memory_growth(gpu, True)
- logical_gpus = tf.config.experimental.list_logical_devices('GPU')
- print(len(gpus), "Physical GPUs,", len(logical_gpus), "Logical GPUs")
- except RuntimeError as e:
- # Memory growth must be set before GPUs have been initialized
- print(e)
+ # gpus = tf.config.experimental.list_physical_devices('GPU')
+ # if gpus:
+ # try:
+ # # Currently, memory growth needs to be the same across GPUs
+ # for gpu in gpus:
+ # tf.config.experimental.set_memory_growth(gpu, True)
+ # logical_gpus = tf.config.experimental.list_logical_devices('GPU')
+ # print(len(gpus), "Physical GPUs,", len(logical_gpus), "Logical GPUs")
+ # except RuntimeError as e:
+ # # Memory growth must be set before GPUs have been initialized
+ # print(e)
def get_in_mem_data_batch(self, idxs, is_training):
--
GitLab