From a97e5ef3f232e022df7b9e9996e0da908b06b482 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 20 Jan 2021 10:37:45 -0600
Subject: [PATCH] snapshot...
---
modules/deeplearning/cloudheight.py | 7 +++----
1 file changed, 3 insertions(+), 4 deletions(-)
diff --git a/modules/deeplearning/cloudheight.py b/modules/deeplearning/cloudheight.py
index 5e9f7fe7..44b7088c 100644
--- a/modules/deeplearning/cloudheight.py
+++ b/modules/deeplearning/cloudheight.py
@@ -250,7 +250,6 @@ class CloudHeightNN:
tf.debugging.set_log_device_placement(LOG_DEVICE_PLACEMENT)
gpus = tf.config.experimental.list_physical_devices('GPU')
- print(gpus)
if gpus:
try:
# Currently, memory growth needs to be the same across GPUs
@@ -845,12 +844,12 @@ class CloudHeightNN:
def run(self, matchup_dict, train_dict=None, valid_dict=None):
#with self.strategy.scope():
- with tf.device('/physical_device:GPU:2'):
+ with tf.device('/device:GPU:'+self.gpu_device):
self.setup_pipeline(matchup_dict, train_dict=train_dict, valid_test_dict=valid_dict)
self.build_model()
self.build_training()
- self.build_evaluation()
- self.do_training()
+ self.build_evaluation()
+ self.do_training()
def run_restore(self, matchup_dict, ckpt_dir):
self.setup_pipeline(None, None, matchup_dict)
--
GitLab