Skip to content

Instantly share code, notes, and snippets.

@bzamecnik

bzamecnik/cudnn_compatible_gru_example.ipynb

Last active November 26, 2021 11:00
Show Gist options
  • Save bzamecnik/bd3786a074f8cb891bc2a397343070f1 to your computer and use it in GitHub Desktop.
Save bzamecnik/bd3786a074f8cb891bc2a397343070f1 to your computer and use it in GitHub Desktop.
Download ZIP
CuDNN-compatible GRU in Keras
Display the source blob
Display the rendered blob
Raw
cudnn_compatible_gru_example.ipynb
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@SatishDivakarla
Copy link

@bzamecnik, Could you please respond to my comment. I have a trained model with CudNNGRU trained on GPU, but I am not able to make them use on CPU environment for predictions. Thanks in advance.

@leowang16
Copy link

Having the same error:
AttributeError: 'Dataset' object has no attribute 'reshape'

@bzamecnik
Copy link
Author

@SatishDivakarla: Sorry, I missed your comment... Don't you have the original stack trace? If I remember correctly I saw something similar in Keras issues and the cause was that the weights were a h5py Dataset instead of a numpy array. I'm not sure about resolution.

@bzamecnik
Copy link
Author

bzamecnik commented Apr 23, 2018 •
edited
Loading

@SatishDivakarla: Aah, keras-team/keras#9112 (comment). Fixed in keras-team/keras#9662 (2018-03-15). Will be released in Keras 2.1.6.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment