From e8b154374a21d692401cf6b62fe9684b2b3827f0 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 20 Jan 2021 11:00:00 -0600
Subject: [PATCH] snapshot...
---
modules/deeplearning/cloudheight.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/modules/deeplearning/cloudheight.py b/modules/deeplearning/cloudheight.py
index 44b7088c..9f28ae68 100644
--- a/modules/deeplearning/cloudheight.py
+++ b/modules/deeplearning/cloudheight.py
@@ -844,7 +844,7 @@ class CloudHeightNN:
def run(self, matchup_dict, train_dict=None, valid_dict=None):
#with self.strategy.scope():
- with tf.device('/device:GPU:'+self.gpu_device):
+ with tf.device('/device:GPU:'+str(self.gpu_device)):
self.setup_pipeline(matchup_dict, train_dict=train_dict, valid_test_dict=valid_dict)
self.build_model()
self.build_training()
--
GitLab