in scripts/adapet/ADAPET/src/utils/Config.py [0:0]
def __init__(self, filename=None, kwargs=None, mkdir=True):
# Dataset parameters
self.dataset = "fewglue/BoolQ"
self.num_lbl = 2
self.max_num_lbl_tok = 10
self.max_num_lbl = 10
# Model and pattern parameters
self.pretrained_weight = "bert-base-uncased"
self.pattern_idx = "random"
# Duration of training parameters
self.batch_size = 8
self.eval_batch_size = 64
self.num_batches = 1000
self.eval_every = 1
self.grad_accumulation_factor = 1
self.max_text_length = 64
self.mask_alpha = 0.5
self.eval_train = False
self.eval_dev = True
# Where experiments will be located
self.exp_dir = None
self.seed = 42
self.exp_name = ""
# Training Hyperparameters
self.lr = 1e-3
self.weight_decay = 0
self.grad_clip_norm = 1
self.warmup_ratio = 0
# Generic dataset hyperparameters
self.pattern = "[TEXT1] and [TEXT2] "
self.idx_txt_trim = -1 # Indexed from 1
self.dict_verbalizer = {"True": "Yes", "False": "No"}
self.data_dir = "data/fewglue/BoolQ"
#Added
self.task_name = 'SetFit/sst2'
if filename:
self.__dict__.update(json.load(open(filename)))
if kwargs:
self.update_kwargs(kwargs)
if filename or kwargs:
self.update_exp_config(mkdir)