Unknown error/crash – TensorFlow LSTM with GPU (no output after start of 1st epoch)
Unknown error/crash – TensorFlow LSTM with GPU (no output after start of 1st epoch) Question: I’m trying to train a model using LSTM layers. I’m using a GPU and all needed libraries are loaded. When I’m building the model this way: model = keras.Sequential() model.add(layers.LSTM(256, activation="relu", return_sequences=False)) # note the activation function model.add(layers.Dropout(0.2)) model.add(layers.Dense(256, activation="relu")) …