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

snapshot...

parent e8b15437
No related branches found
No related tags found
No related merge requests found
......@@ -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()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment