diff --git a/modules/deeplearning/icing.py b/modules/deeplearning/icing.py index f960e9e125378128e4bf10a3f042ee6c72b57c69..1eb1634a2d33e0dce195d6af0e10fc860050aaa6 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()