We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7dd06ca commit 2b198edCopy full SHA for 2b198ed
tests/data/cifar_10/source/resnet_cifar_10.py
@@ -33,8 +33,8 @@
33
BATCH_SIZE = 1
34
35
# Scale the learning rate linearly with the batch size. When the batch size is
36
-# 128, the learning rate should be 0.1.
37
-_INITIAL_LEARNING_RATE = 0.1 * BATCH_SIZE / 128
+# 128, the learning rate should be 0.05.
+_INITIAL_LEARNING_RATE = 0.05 * BATCH_SIZE / 128
38
_MOMENTUM = 0.9
39
40
# We use a weight decay of 0.0002, which performs better than the 0.0001 that
0 commit comments