Skip to content

Commit d5f5c9d

Browse files
committed
Update to add global norm
1 parent f9063db commit d5f5c9d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Diff for: train.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -17,9 +17,9 @@
1717
NUM_LAYERS = 4 # number of layers of the state space
1818
MAX_TRIALS = 250 # maximum number of models generated
1919

20-
MAX_EPOCHS = 1 # maximum number of epochs to train
20+
MAX_EPOCHS = 10 # maximum number of epochs to train
2121
CHILD_BATCHSIZE = 128 # batchsize of the child models
22-
EXPLORATION = 0.9 # high exploration for the first 1000 steps
22+
EXPLORATION = 0.8 # high exploration for the first 1000 steps
2323
REGULARIZATION = 1e-3 # regularization strength
2424
CONTROLLER_CELLS = 32 # number of cells in RNN controller
2525
EMBEDDING_DIM = 20 # dimension of the embeddings for each state

0 commit comments

Comments
 (0)