We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f9063db commit d5f5c9dCopy full SHA for d5f5c9d
train.py
@@ -17,9 +17,9 @@
17
NUM_LAYERS = 4 # number of layers of the state space
18
MAX_TRIALS = 250 # maximum number of models generated
19
20
-MAX_EPOCHS = 1 # maximum number of epochs to train
+MAX_EPOCHS = 10 # maximum number of epochs to train
21
CHILD_BATCHSIZE = 128 # batchsize of the child models
22
-EXPLORATION = 0.9 # high exploration for the first 1000 steps
+EXPLORATION = 0.8 # high exploration for the first 1000 steps
23
REGULARIZATION = 1e-3 # regularization strength
24
CONTROLLER_CELLS = 32 # number of cells in RNN controller
25
EMBEDDING_DIM = 20 # dimension of the embeddings for each state
0 commit comments