-
Notifications
You must be signed in to change notification settings - Fork 183
Description
rzai@rzai00:/prj/visual-qa/scripts$ CUDA_VISIBLE_DEVICES=1 python trainLSTM_language.py/prj/visual-qa/scripts$
Using Theano backend.
Using gpu device 0: GeForce GTX 1080 (CNMeM is disabled, cuDNN 5105)
Loaded questions, sorting by length...
Compiling model...
Compilation done...
loaded word2vec features...
Training started...
Traceback (most recent call last):
File "trainLSTM_language.py", line 91, in
main()
File "trainLSTM_language.py", line 81, in main
loss = model.train_on_batch(X_q_batch, Y_batch)
File "/usr/local/lib/python2.7/dist-packages/keras/models.py", line 712, in train_on_batch
class_weight=class_weight)
File "/usr/local/lib/python2.7/dist-packages/keras/engine/training.py", line 1215, in train_on_batch
check_batch_dim=True)
File "/usr/local/lib/python2.7/dist-packages/keras/engine/training.py", line 961, in _standardize_user_data
exception_prefix='model input')
File "/usr/local/lib/python2.7/dist-packages/keras/engine/training.py", line 108, in standardize_input_data
str(array.shape))
Exception: Error when checking model input: expected lstm_input_1 to have shape (None, 30, 300) but got array with shape (128, 5, 300)
rzai@rzai00: