some updates

2019-05-18 18:49:40 +08:00
parent ec5c4ac97c
commit f136b8f60a
4 changed files with 28 additions and 7 deletions
--- a/tools/TrainCNN/backward.py
+++ b/tools/TrainCNN/backward.py
@@ -75,7 +75,7 @@ def train(dataset, show_bar=False):
    learning_rate = tf.train.exponential_decay(
        LEARNING_RATE_BASE,
        global_step,
-        len(dataset.train_samples) / BATCH,
+        len(dataset.train_samples) / BATCH / 5,
        LEARNING_RATE_DECAY,
        staircase=False)
    train_step = tf.train.AdamOptimizer(learning_rate).minimize(loss, global_step=global_step)