diff --git a/modules/deeplearning/icing.py b/modules/deeplearning/icing.py
index 2d9924db1802ccff2eac411bd069bf71b8b82f32..7aecfdbaf93eb2669aa55fe5e66ee9b6c22d2630 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()