snapshot...

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()