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

snapshot...

parent e472a87b
Branches sphinx
No related tags found
No related merge requests found
......@@ -638,11 +638,11 @@ class SRCNN:
return pred
def run(self, directory, ckpt_dir=None):
def run(self, directory, ckpt_dir=None, num_data_samples=50000):
train_data_files = glob.glob(directory+'data_train*.npy')
valid_data_files = glob.glob(directory+'data_valid*.npy')
self.setup_pipeline(train_data_files, valid_data_files, 50000)
self.setup_pipeline(train_data_files, valid_data_files, num_data_samples)
self.build_model()
self.build_training()
self.build_evaluation()
......
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