diff --git a/modules/deeplearning/cloudheight.py b/modules/deeplearning/cloudheight.py
index c4562315b0b87e92587f9ccbb9a678dfc22eaeb6..ae3bad25935c4bc5568fc71d9afefdfd51cc925d 100644
--- a/modules/deeplearning/cloudheight.py
+++ b/modules/deeplearning/cloudheight.py
@@ -843,12 +843,12 @@ class CloudHeightNN:
         print('acc_5', self.num_5, self.accuracy_5.result())
 
     def run(self, matchup_dict, train_dict=None, valid_dict=None):
-        with self.strategy.scope():
-            self.setup_pipeline(matchup_dict, train_dict=train_dict, valid_test_dict=valid_dict)
-            self.build_model()
-            self.build_training()
-            self.build_evaluation()
-            self.do_training()
+        #with self.strategy.scope():
+        self.setup_pipeline(matchup_dict, train_dict=train_dict, valid_test_dict=valid_dict)
+        self.build_model()
+        self.build_training()
+        self.build_evaluation()
+        self.do_training()
 
     def run_restore(self, matchup_dict, ckpt_dir):
         self.setup_pipeline(None, None, matchup_dict)