From 6cd9445e1727e61307167c6183afe009b91c8b0c Mon Sep 17 00:00:00 2001
From: coolneng <akasroua@gmail.com>
Date: Tue, 6 Jul 2021 19:04:53 +0200
Subject: [PATCH] Remove batch size from the Input layer

---
 src/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/model.py b/src/model.py
index 39f7ab4..413b683 100644
--- a/src/model.py
+++ b/src/model.py
@@ -20,7 +20,7 @@ def build_model(hyperparams) -> Model:
     """
     model = Sequential(
         [
-            Input(shape=(hyperparams.batch_size, hyperparams.max_length, len(BASES))),
+            Input(shape=(hyperparams.max_length, len(BASES))),
             Masking(mask_value=-1),
             Dense(
                 units=256, activation="relu", kernel_regularizer=l2(hyperparams.l2_rate)