coolneng
/
locimend
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add learning rate and l2 regularizer constants

This commit is contained in:
coolneng 2021-06-03 18:52:26 +02:00
parent ccaa8484c7
commit c9de0c8320
Signed by: coolneng
GPG Key ID: 9893DA236405AF57
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/constants.py
View File

@ -3,3 +3,5 @@ TRAIN_DATASET = "data/train_data.tfrecords"
TEST_DATASET = "data/test_data.tfrecords" TEST_DATASET = "data/test_data.tfrecords"
EPOCHS = 1000 EPOCHS = 1000
BATCH_SIZE = 256 BATCH_SIZE = 256
LEARNING_RATE = 0.004
L2 = 0.001