From bc220dfdcd29c952e3f6fff7e0709d4b519d4442 Mon Sep 17 00:00:00 2001
From: vetterjn <25137666+vetterjn@users.noreply.github.com>
Date: Wed, 11 Dec 2019 14:45:02 +0100
Subject: [PATCH] Update data_loader.py

---
 data_loader.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/data_loader.py b/data_loader.py
index e31715c..0c22d73 100644
--- a/data_loader.py
+++ b/data_loader.py
@@ -11,7 +11,7 @@ class VqaDataset(data.Dataset):
 
     def __init__(self, input_dir, input_vqa, max_qst_length=30, max_num_ans=10, transform=None):
         self.input_dir = input_dir
-        self.vqa = np.load(input_dir+'/'+input_vqa)
+        self.vqa = np.load(input_dir+'/'+input_vqa, allow_pickle=True)
         self.qst_vocab = text_helper.VocabDict(input_dir+'/vocab_questions.txt')
         self.ans_vocab = text_helper.VocabDict(input_dir+'/vocab_answers.txt')
         self.max_qst_length = max_qst_length