Skip to content
Snippets Groups Projects
Commit 070f4906 authored by tomrink's avatar tomrink
Browse files

snapshot...

parent e8b15437
Branches
No related tags found
No related merge requests found
...@@ -261,8 +261,6 @@ class CloudHeightNN: ...@@ -261,8 +261,6 @@ class CloudHeightNN:
# Memory growth must be set before GPUs have been initialized # Memory growth must be set before GPUs have been initialized
print(e) print(e)
#self.strategy = tf.distribute.MirroredStrategy()
def get_in_mem_data_batch(self, time_keys): def get_in_mem_data_batch(self, time_keys):
images = [] images = []
vprof = [] vprof = []
...@@ -843,7 +841,6 @@ class CloudHeightNN: ...@@ -843,7 +841,6 @@ class CloudHeightNN:
print('acc_5', self.num_5, self.accuracy_5.result()) print('acc_5', self.num_5, self.accuracy_5.result())
def run(self, matchup_dict, train_dict=None, valid_dict=None): def run(self, matchup_dict, train_dict=None, valid_dict=None):
#with self.strategy.scope():
with tf.device('/device:GPU:'+str(self.gpu_device)): with tf.device('/device:GPU:'+str(self.gpu_device)):
self.setup_pipeline(matchup_dict, train_dict=train_dict, valid_test_dict=valid_dict) self.setup_pipeline(matchup_dict, train_dict=train_dict, valid_test_dict=valid_dict)
self.build_model() self.build_model()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment