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

snapshot..

parent 9eb828f2
Branches
No related tags found
No related merge requests found
...@@ -694,7 +694,7 @@ class SRCNN: ...@@ -694,7 +694,7 @@ class SRCNN:
self.do_training(ckpt_dir=ckpt_dir) self.do_training(ckpt_dir=ckpt_dir)
def run_restore(self, directory, ckpt_dir): def run_restore(self, directory, ckpt_dir):
valid_data_files = glob.glob(directory + 'data_*.npy') valid_data_files = glob.glob(directory + 'data_valid*.npy')
self.num_data_samples = 1000 self.num_data_samples = 1000
self.setup_test_pipeline(valid_data_files) self.setup_test_pipeline(valid_data_files)
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