From 120b591c9be5426fd2cc0df04c2b5a93bc15cf89 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 19 Jan 2021 20:28:50 -0600
Subject: [PATCH] snapshot...
---
modules/deeplearning/cloudheight.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/modules/deeplearning/cloudheight.py b/modules/deeplearning/cloudheight.py
index b98c781c..5e9f7fe7 100644
--- a/modules/deeplearning/cloudheight.py
+++ b/modules/deeplearning/cloudheight.py
@@ -845,9 +845,10 @@ class CloudHeightNN:
def run(self, matchup_dict, train_dict=None, valid_dict=None):
#with self.strategy.scope():
- self.setup_pipeline(matchup_dict, train_dict=train_dict, valid_test_dict=valid_dict)
- self.build_model()
- self.build_training()
+ with tf.device('/physical_device:GPU:2'):
+ 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()
--
GitLab