Commit
·
7863a65
1
Parent(s):
1bf74fe
update
Browse files- configuration_bbb.py +2 -2
configuration_bbb.py
CHANGED
|
@@ -9,8 +9,8 @@ class BBBConfig(PretrainedConfig):
|
|
| 9 |
d_img: int = 2048,
|
| 10 |
d_txt: int = 768,
|
| 11 |
proj_dim: int = 2048,
|
| 12 |
-
dropout: float,
|
| 13 |
-
task: str,
|
| 14 |
**kwargs):
|
| 15 |
|
| 16 |
self.d_tab = d_tab
|
|
|
|
| 9 |
d_img: int = 2048,
|
| 10 |
d_txt: int = 768,
|
| 11 |
proj_dim: int = 2048,
|
| 12 |
+
dropout: float = 0.1,
|
| 13 |
+
task: str = 'classification',
|
| 14 |
**kwargs):
|
| 15 |
|
| 16 |
self.d_tab = d_tab
|