-
Notifications
You must be signed in to change notification settings - Fork 1
/
CONFIG.py
36 lines (33 loc) · 1.24 KB
/
CONFIG.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
class CONFIG:
def __init__(self):
self.log_path = "log.txt"
self.tokenizer_path = "t5-base"
self.model_config_path = "config/model_config.json"
self.device = "gpu"
self.saved_model_path = "output_ultra-fine"
self.pretrained_model_path = "t5-base"
self.test = True
self.train = False
self.num_workers = 0
self.train_data_path = 'dataset/Ultra-fine_all.json'
#self.dev_data_path = 'dataset/mask_dev_split_dataset_not_prompt.json'
self.test_data_path = 'dataset/mask_split_test_ultra_sample.json'
self.test_size = 0.1
self.batch_size= 48
self.epochs = 10
self.save_model_dic="output_ultra-fine/model_epoch10/Prompt_IsA_Model.pth"
self.c1="output_ultra-fine/model_epoch10/Prompt_IsA_Model.pth"
self.lr=1.5e-4
self.warmup_steps=2000
self.log_step=1
self.max_grad_norm=1.0
self.gradient_accumulation=1
self.topk=8
self.test_max_len=20
self.save_mode=True
self.n_desc=128
self.n_labels=16
self.num_classification=19
self.together=False
self.result_text = "result_ultra-fine.txt"
self.result_text_c2 = "result_ultra-fine.txt"