-
Notifications
You must be signed in to change notification settings - Fork 6.8k
Conversation
@pluskid |
@piiswrong yes agreed. The currently used LSTM in ptb and speech are a variant with some simplification. An implementation of a full version needs the broadcast multiplication feature. Need to check what variant of the LSTM cell is cuDNN v5 providing. What is our current status of supporting cuDNN v5? I am reluctant to upgrade having heard that they breaks a lot of APIs.... also on servers I do not have permission to upgrade cuda driver to v8. |
@pluskid We now support everything but LSTM cell in CuDNN v5. CUDA 7.5 is fine for CuDNN v5. |
@@ -28,33 +28,33 @@ export NVCC = nvcc | |||
DEBUG = 0 | |||
|
|||
# the additional link flags you want to add | |||
ADD_LDFLAGS = | |||
ADD_LDFLAGS = -L/disk1/deeplearning/local_install/lib |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This file should not be modified.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
thanks, I will modify it
Glad to hear that mxnet will have a full support of LSTM. Another think I want to ask is, is it easy to integrate baidu wrap-ctc without using torch module? |
this is ready to be merged after combine the commits. see instructions http://mxnet.readthedocs.io/en/latest/how_to/contribute.html#how-to-combine-multiple-commits-into-one |
sorry to response later. I combined the commits but find its already merged. Thanks all. |
@xlvector I am working on make this example python2/3 compatible. However, I find that the data folder and gen_data.py are missing in the current master branch. Could you take a look at this? |
Thanks. The gen_data.py is very simple, it just generate whitespace seprated numbers in every line. something like for i in range(1000): You can add this script in your PR. |
ok I will put this gen_data into the lstm_sort.py directly. Thanks. |
This is an just simple example for bidirectional lstm.