Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -63,11 +63,11 @@ def create_network(network_input, n_vocab):
|
|
63 |
))
|
64 |
model.add(LSTM(512, return_sequences=True, recurrent_dropout=0.3,))
|
65 |
model.add(LSTM(256))
|
66 |
-
model.add(
|
67 |
model.add(Dropout(0.2))
|
68 |
model.add(Dense(256))
|
69 |
model.add(Activation('relu'))
|
70 |
-
model.add(
|
71 |
model.add(Dropout(0.2))
|
72 |
model.add(Dense(n_vocab))
|
73 |
model.add(Activation('softmax'))
|
|
|
63 |
))
|
64 |
model.add(LSTM(512, return_sequences=True, recurrent_dropout=0.3,))
|
65 |
model.add(LSTM(256))
|
66 |
+
model.add(BatchNormalization())
|
67 |
model.add(Dropout(0.2))
|
68 |
model.add(Dense(256))
|
69 |
model.add(Activation('relu'))
|
70 |
+
model.add(BatchNormalization())
|
71 |
model.add(Dropout(0.2))
|
72 |
model.add(Dense(n_vocab))
|
73 |
model.add(Activation('softmax'))
|