avisingh599 / visual-qa

[Reimplementation Antol et al 2015] Keras-based LSTM/CNN models for Visual Question Answering
https://avisingh599.github.io/deeplearning/visual-qa/
MIT License
480 stars 186 forks source link

expected lstm_input_1 to have shape (None, 30, 300) but got array with shape (128, 5, 300) #29

Open SeekPoint opened 7 years ago

SeekPoint commented 7 years ago

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$