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

snapshot...

parent f23dca45
Branches
No related tags found
No related merge requests found
......@@ -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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment