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

snapshot...

parent 8cd2ae58
Branches
No related tags found
No related merge requests found
......@@ -467,7 +467,7 @@ class SRCNN:
self.loss = tf.keras.losses.MeanSquaredError() # Regression
# decayed_learning_rate = learning_rate * decay_rate ^ (global_step / decay_steps)
initial_learning_rate = 0.002
initial_learning_rate = 0.001
decay_rate = 0.95
steps_per_epoch = int(self.num_data_samples/BATCH_SIZE) # one epoch
decay_steps = int(steps_per_epoch) * 4
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment