batra-mlp-lab / visdial-rl

PyTorch code for Learning Cooperative Visual Dialog Agents using Deep Reinforcement Learning
170 stars 39 forks source link

Error Creating data mats for train.... #15

Closed abrar1515 closed 4 years ago

abrar1515 commented 4 years ago

kindly help me error is occur during executing of prepro.py file error is Traceback (most recent call last): File "prepro.py", line 274, in captions_train, captions_train_len, questions_train, questions_train_len, answers_train, answers_train_len, options_train, options_train_list, options_train_len, answers_train_index, images_train_index, images_trainlist, = create_data_mats(data_train_toks, ques_train_inds, ans_train_inds, args, 'train') File "prepro.py", line 127, in create_data_mats options[i][j] = np.array(data_toks[image_id]['dialog'][j]['answer_options']) + 1 ValueError: could not broadcast input array from shape (104) into shape (100)

abrar1515 commented 4 years ago

solved with online downloading...