Yes, import GRU from keras.layers (instead of CuDNNGRU) and replace all instances of CuDNNGRU with GRU. This allows you to train the neural network using a CPU.
Written by
Yes, import GRU from keras.layers (instead of CuDNNGRU) and replace all instances of CuDNNGRU with GRU. This allows you to train the neural network using a CPU.
Written by
