TensorFlow застревает после использования сцепленного слоя - PullRequest
Новая
       20

TensorFlow застревает после использования сцепленного слоя

0 голосов
/ 03 апреля 2020

У меня есть следующая модель: 

import tensorflow as tf

input1 = tf.keras.layers.Input(shape = (10, 300, 1))
input2 = tf.keras.layers.Input(shape = (24, ))
x = tf.keras.layers.Conv2D(64, (3,3), activation='relu')(input1)

x = tf.keras.layers.MaxPooling2D(2,2)(x)
x = tf.keras.layers.Dropout(0.25)(x)
x = tf.keras.layers.Conv2D(128, (2,2), activation='relu')(x)
x = tf.keras.layers.MaxPooling2D(2,2)(x)
x = tf.keras.layers.Dropout(0.25)(x)
x = tf.keras.layers.Flatten()(x)
x = tf.keras.layers.Dense(512, activation = 'relu')(x)
x = tf.keras.layers.Dropout(0.25)(x)
x = tf.keras.layers.Concatenate()([x, input2])
x = tf.keras.layers.Dense(128, activation = 'relu')(x)
x = tf.keras.layers.Dropout(0.25)(x)
output = tf.keras.layers.Dense(1, activation='sigmoid')(x)

model = tf.keras.models.Model(inputs = [input1,input2], outputs = output)

model.summary()
model.compile(optimizer = 'rmsprop',
          loss ='binary_crossentropy',
          metrics = ['acc'])

history = model.fit([X_train, X_features], y_train,
                batch_size=64,
                epochs=100)

Но когда я пытаюсь подогнать его, застреваю и появляется только Эпоха 1/100, и больше ничего не происходит, даже если я позволяю ему работать часами. Но когда я удаляю слой сцепления, все go хорошо. Я использую Google Colab. Почему это происходит?

...