update

classification/configuration_bbb.py

@@ -0,0 +1,22 @@
+from transformers import PretrainedConfig
+
+class BBBConfig(PretrainedConfig):
+  model_type = "bbb-model"
+
+  def __init__(
+    self,
+    d_tab : int = 384, 
+    d_img: int = 2048, 
+    d_txt: int = 768, 
+    proj_dim: int = 2048, 
+    dropout: float = 0.1, 
+    task: str = 'classification',
+    **kwargs):
+
+    self.d_tab = d_tab
+    self.d_img = d_img
+    self.d_txt = d_txt
+    self.proj_dim = proj_dim
+    self.dropout = dropout
+    self.task = task
+    super().__init__(**kwargs)