Commit 59bbe025 authored by sjromuel's avatar sjromuel
Browse files

d

parent 98f45f9a
......@@ -115,7 +115,7 @@ def main():
lr = 1e-4
optimizer = tf.optimizers.Adam(learning_rate=lr)
batch_size = 5
num_epochs = 50
num_epochs = 100
augment = ['normal', 'fliplr', 'flipud', 'rotate']
verbose = True
save_weights = True
......
......@@ -114,7 +114,7 @@ def main():
lr = 1e-4
optimizer = tf.optimizers.Adam(learning_rate=lr)
batch_size = 5
num_epochs = 50
num_epochs = 100
augment = ['normal', 'fliplr', 'flipud', 'rotate']
verbose = True
save_weights = True
......
......@@ -112,7 +112,7 @@ def main():
lr = 1e-4
optimizer = tf.optimizers.Adam(learning_rate=lr)
batch_size = 5
num_epochs = 50
num_epochs = 100
augment = ['normal', 'fliplr', 'flipud', 'rotate']
verbose = True
save_weights = True
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment