From 6475cf591126979689a90229e58dec90ad6f19f3 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 19 Apr 2021 17:03:44 -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 2d9924db..7aecfdba 100644
--- a/modules/deeplearning/icing.py
+++ b/modules/deeplearning/icing.py
@@ -567,7 +567,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