update

classification/modeling_bbb.py

@@ -4,7 +4,6 @@ import torch.nn.functional as F
 
 from transformers import PreTrainedModel
 from transformers.modeling_outputs import BaseModelOutputWithPooling, SequenceClassifierOutput
-import os
 
 from .configuration_bbb import BBBConfig
 

configuration_bbb.py

@@ -9,8 +9,8 @@ class BBBConfig(PretrainedConfig):
     d_img: int = 2048, 
     d_txt: int = 768, 
     proj_dim: int = 2048, 
-    dropout: float = 0.1, 
-    task: str = 'classification',
+    dropout: float, 
+    task: str,
     **kwargs):
 
     self.d_tab = d_tab

regression/config.json

@@ -3,8 +3,8 @@
     "BBBModelForSequenceClassification"
   ],
   "auto_map": {
-    "AutoConfig": "configuration_bbb.BBBConfig",
-    "AutoModelForSequenceClassification": "modeling_bbb.BBBModelForSequenceClassification"
+    "AutoConfig": "regression/configuration_bbb.BBBConfig",
+    "AutoModelForSequenceClassification": "regression/modeling_bbb.BBBModelForSequenceClassification"
   },
   "d_img": 2048,
   "d_tab": 384,