Open milkmilkMilktea opened 5 years ago
This network can only be trained on a GPU because it uses CuDNNGRU, the error sounds like you're trying to run it on a CPU. You may be able to fix the problem by replacing that layer with a regular GRU.
Thank you so so much @HackerPoet how would I run it on my GPU? In the past I've just used Google Colab, but there must be another way?
woops I meant to link something, guess it didn't work https://www.codingforentrepreneurs.com/blog/install-tensorflow-gpu-windows-cuda-cudnn/
Installing CUDA and cuDNN is best because it will train significantly faster. But if you can't get it working, just replace the instance of 'CuDNNGRU' with 'GRU'.
After installing 5 different versions of CUDA and cuDNN, I just gave up and used GRU and it worked great! By chance could you tell me what versions you use for future reference?
Make sure you have Nvidia graphics card as CUDA & cuDNN are Nvidia's library.
So everything runs fine until the line:
batch_loss, batch_acc = model.train_on_batch([title_batch, past_batch], [pred_batch])
I figured it was a syntax error or something so I did some research. Although I've didn't see it used this way, there was nothing wrong with the syntax. So I updated all of my Libraries and tried it again. Still the same error. No matter what I did, I couldn't get it to work. If anyone cares, or is even out there (this is not a maintained repository so maybe I'm alone), your help would be immensely appreciated.
Here's the full error log. pls help me