Add learning rate and l2 regularizer constants

2021-06-03 18:52:26 +02:00 · 2021-06-03 18:52:26 +02:00 · c9de0c8320
parent ccaa8484c7
commit c9de0c8320
1 changed files with 2 additions and 0 deletions
--- a/src/constants.py
+++ b/src/constants.py
@ -3,3 +3,5 @@ TRAIN_DATASET = "data/train_data.tfrecords"
 TEST_DATASET = "data/test_data.tfrecords"
 EPOCHS = 1000
 BATCH_SIZE = 256
+LEARNING_RATE = 0.004
+L2 = 0.001