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:
self.do_training(ckpt_dir=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.setup_test_pipeline(valid_data_files)
self.build_model()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment