Change model architecture to a MLP

2021-07-06 01:44:58 +02:00
parent 1a1262b0b1
commit eabb7f0285
2 changed files with 9 additions and 16 deletions
--- a/src/hyperparameters.py
+++ b/src/hyperparameters.py
@@ -6,8 +6,8 @@ class Hyperparameters:
        train_dataset="data/train_data.tfrecords",
        test_dataset="data/test_data.tfrecords",
        eval_dataset="data/eval_data.tfrecords",
-        epochs=1000,
-        batch_size=256,
+        epochs=100,
+        batch_size=64,
        learning_rate=0.004,
        l2_rate=0.001,
        log_directory="logs",