From 61356bd7c7c5c5d300f0ec48cb8535002c061c1b Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 19 Apr 2021 11:14:58 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/modules/deeplearning/icing.py b/modules/deeplearning/icing.py
index f960e9e1..1eb1634a 100644
--- a/modules/deeplearning/icing.py
+++ b/modules/deeplearning/icing.py
@@ -562,7 +562,7 @@ class IcingIntensityNN:
print('loss, acc: ', self.test_loss.result(), self.test_accuracy.result())
def run(self, filename, filename_l1b=None, train_dict=None, valid_dict=None):
- with tf.device('/device:GPU:'+str(self.gpu_device)):
+ #with tf.device('/device:GPU:'+str(self.gpu_device)):
self.setup_pipeline(filename, train_idxs=train_dict, test_idxs=valid_dict)
self.build_model()
self.build_training()
--
GitLab