diff --git a/modules/deeplearning/cloudheight.py b/modules/deeplearning/cloudheight.py index 9f28ae68283b3719ddf275f1925bbc979961c1d3..eacd2fc33a7f94c3796bd85d645981ffa16b782f 100644 --- a/modules/deeplearning/cloudheight.py +++ b/modules/deeplearning/cloudheight.py @@ -261,8 +261,6 @@ class CloudHeightNN: # Memory growth must be set before GPUs have been initialized print(e) - #self.strategy = tf.distribute.MirroredStrategy() - def get_in_mem_data_batch(self, time_keys): images = [] vprof = [] @@ -843,7 +841,6 @@ 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(): with tf.device('/device:GPU:'+str(self.gpu_device)): self.setup_pipeline(matchup_dict, train_dict=train_dict, valid_test_dict=valid_dict) self.build_model()