ierhon commited on
Commit
f6ba3ec
1 Parent(s): 147d2ba

Fix minimum input length

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -33,8 +33,8 @@ def train(message: str, epochs: int, learning_rate: float, emb_size: int, inp_le
33
  attn_layer = MultiHeadAttention(num_heads=4, key_dim=128)(emb_layer, emb_layer, emb_layer)
34
  noise_layer = GaussianNoise(0.1)(attn_layer)
35
  conv1_layer = Conv1D(64, 8, padding='same', activation='relu', strides=1, input_shape=(64, 128))(noise_layer)
36
- conv2_layer = Conv1D(16, 4, padding='valid', activation='relu', strides=1)(conv1_layer)
37
- conv3_layer = Conv1D(8, 2, padding='valid', activation='relu', strides=1)(conv2_layer)
38
  flatten_layer = Flatten()(conv3_layer)
39
  attn_flatten_layer = Flatten()(attn_layer)
40
  conv1_flatten_layer = Flatten()(conv1_layer)
 
33
  attn_layer = MultiHeadAttention(num_heads=4, key_dim=128)(emb_layer, emb_layer, emb_layer)
34
  noise_layer = GaussianNoise(0.1)(attn_layer)
35
  conv1_layer = Conv1D(64, 8, padding='same', activation='relu', strides=1, input_shape=(64, 128))(noise_layer)
36
+ conv2_layer = Conv1D(16, 4, padding='same', activation='relu', strides=1)(conv1_layer)
37
+ conv3_layer = Conv1D(8, 2, padding='same', activation='relu', strides=1)(conv2_layer)
38
  flatten_layer = Flatten()(conv3_layer)
39
  attn_flatten_layer = Flatten()(attn_layer)
40
  conv1_flatten_layer = Flatten()(conv1_layer)