Update tester.py
Browse files
tester.py
CHANGED
@@ -162,7 +162,7 @@ class CaptionSampler(object):
|
|
162 |
return ' '.join(sampled_caption)
|
163 |
|
164 |
def __init_vocab(self):
|
165 |
-
with open(
|
166 |
vocab = pickle.load(f)
|
167 |
return vocab
|
168 |
|
|
|
162 |
return ' '.join(sampled_caption)
|
163 |
|
164 |
def __init_vocab(self):
|
165 |
+
with open('vocab.pkl', 'rb') as f:
|
166 |
vocab = pickle.load(f)
|
167 |
return vocab
|
168 |
|