eec61d9d4921fb7a62926e217fbfe43cd7062ac4,examples/lstm_benchmark.py,,,#,28

Before Change

embedding_dim = 256
batch_size = 128
epochs = 10
modes = ["cpu", "mem", "gpu"]

print("Loading data...")
(X_train, y_train), (X_test, y_test) = imdb.load_data(num_words=max_features)
X_train = sequence.pad_sequences(X_train, max_length)

After Change

    model = Sequential()
    model.add(Embedding(max_features, embedding_dim, input_length=max_length))
    model.add(Dropout(0.2))
    model.add(LSTM(embedding_dim, dropout=0.2, recurrent_dropout=0.2, implementation=mode))
    model.add(Dense(1, activation="sigmoid"))
    model.compile(loss="binary_crossentropy",
                  optimizer="adam",

Instances