diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py index 13270c745e67c1e156507804a2221c5328f92777..1b268c36bd2935c4dfb7ea977c73fe3e778721b2 100644 --- a/modules/deeplearning/icing_fcn.py +++ b/modules/deeplearning/icing_fcn.py @@ -955,8 +955,6 @@ class IcingIntensityFCN: self.test_preds = preds def run(self, filename_l1b_trn, filename_l1b_tst, filename_l2_trn, filename_l2_tst): - # This doesn't really play well with SLURM - # with tf.device('/device:GPU:'+str(self.gpu_device)): self.setup_pipeline(filename_l1b_trn, filename_l1b_tst, filename_l2_trn, filename_l2_tst) self.build_model() self.build_training()