Commit
•
a77dcc0
1
Parent(s):
0b8acb0
Update ViTCoAtt.py
Browse files- ViTCoAtt.py +1 -1
ViTCoAtt.py
CHANGED
|
@@ -18,7 +18,7 @@ class CaptionSampler(object):
|
|
| 18 |
def __init__(self):
|
| 19 |
# Default configuration values
|
| 20 |
self.args = {
|
| 21 |
-
"model_dir": "",
|
| 22 |
"image_dir": "",
|
| 23 |
"caption_json": "",
|
| 24 |
"vocab_path": "vocab.pkl",
|
|
|
|
| 18 |
def __init__(self):
|
| 19 |
# Default configuration values
|
| 20 |
self.args = {
|
| 21 |
+
"model_dir": "model/",
|
| 22 |
"image_dir": "",
|
| 23 |
"caption_json": "",
|
| 24 |
"vocab_path": "vocab.pkl",
|