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