Spaces:
Runtime error
Runtime error
undo lr change
Browse files- pipelines/train.py +1 -1
pipelines/train.py
CHANGED
@@ -21,7 +21,7 @@ from cnn import CNNetwork
|
|
21 |
# script defaults
|
22 |
BATCH_SIZE = 128
|
23 |
EPOCHS = 10
|
24 |
-
LEARNING_RATE = 0.
|
25 |
|
26 |
TRAIN_FILE="data/train"
|
27 |
TEST_FILE="data/test"
|
|
|
21 |
# script defaults
|
22 |
BATCH_SIZE = 128
|
23 |
EPOCHS = 10
|
24 |
+
LEARNING_RATE = 0.001
|
25 |
|
26 |
TRAIN_FILE="data/train"
|
27 |
TEST_FILE="data/test"
|