ierhon commited on
Commit
7b5b35f
1 Parent(s): ebc3421

Fix concatenate incorrect shape error

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -39,8 +39,9 @@ def train(data: str, message: str):
39
  flatten_layer = Flatten()(conv3_layer)
40
  attn_flatten_layer = Flatten()(attn_layer)
41
  conv1_flatten_layer = Flatten()(conv1_layer)
 
42
  conv3_flatten_layer = Flatten()(conv3_layer)
43
- concat1_layer = Concatenate()([flatten_layer, attn_flatten_layer, conv1_flatten_layer, conv2_layer, conv3_flatten_layer])
44
  dense1_layer = Dense(512, activation="linear")(concat1_layer)
45
  prelu1_layer = PReLU()(dense1_layer)
46
  dropout_layer = Dropout(0.3)(prelu1_layer)
 
39
  flatten_layer = Flatten()(conv3_layer)
40
  attn_flatten_layer = Flatten()(attn_layer)
41
  conv1_flatten_layer = Flatten()(conv1_layer)
42
+ conv2_flatten_layer = Flatten()(conv2_layer)
43
  conv3_flatten_layer = Flatten()(conv3_layer)
44
+ concat1_layer = Concatenate()([flatten_layer, attn_flatten_layer, conv1_flatten_layer, conv2_flatten_layer, conv3_flatten_layer])
45
  dense1_layer = Dense(512, activation="linear")(concat1_layer)
46
  prelu1_layer = PReLU()(dense1_layer)
47
  dropout_layer = Dropout(0.3)(prelu1_layer)