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
|