Update implement.py
Browse files- implement.py +1 -1
implement.py
CHANGED
@@ -16,7 +16,7 @@ import os
|
|
16 |
class CFG:
|
17 |
debug = False
|
18 |
image_path = ""
|
19 |
-
captions_path =
|
20 |
batch_size = 30
|
21 |
num_workers = 4
|
22 |
head_lr = 1e-3
|
|
|
16 |
class CFG:
|
17 |
debug = False
|
18 |
image_path = ""
|
19 |
+
captions_path = os.getcwd()
|
20 |
batch_size = 30
|
21 |
num_workers = 4
|
22 |
head_lr = 1e-3
|