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

Remove batch size from the Input layer

This commit is contained in:
coolneng 2021-07-06 19:04:53 +02:00
parent acd231e633
commit 6cd9445e17
Signed by: coolneng
GPG Key ID: 9893DA236405AF57
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/model.py
View File

@ -20,7 +20,7 @@ def build_model(hyperparams) -> Model:
""" """
model = Sequential( model = Sequential(
[ [
Input(shape=(hyperparams.batch_size, hyperparams.max_length, len(BASES))), Input(shape=(hyperparams.max_length, len(BASES))),
Masking(mask_value=-1), Masking(mask_value=-1),
Dense( Dense(
units=256, activation="relu", kernel_regularizer=l2(hyperparams.l2_rate) units=256, activation="relu", kernel_regularizer=l2(hyperparams.l2_rate)