rzai@rzai00:~/prj/visual-qa/scripts$ CUDA_VISIBLE_DEVICES=1 python trainLSTM_language.py
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:~/prj/visual-qa/scripts$
rzai@rzai00:~/prj/visual-qa/scripts$ CUDA_VISIBLE_DEVICES=1 python trainLSTM_language.py 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:~/prj/visual-qa/scripts$