From c9de0c83207a66057c5e0c46d23a354c099d004a Mon Sep 17 00:00:00 2001 From: coolneng Date: Thu, 3 Jun 2021 18:52:26 +0200 Subject: [PATCH] Add learning rate and l2 regularizer constants --- src/constants.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/constants.py b/src/constants.py index a9be1d0..c44726b 100644 --- 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