# coding:utf-8
import torch


class ProjectConfig(object):
    def __init__(self):
        self.device = 'cuda:0' if torch.cuda.is_available() else 'cpu'
        self.pre_model = r'model\bert-base-chinese'
        self.train_path = r'data2/train4.txt'
        self.dev_path = r'data2/dev4.txt'
        self.verbalizer = r'data/verbalizer.txt'
        self.max_seq_len = 512
        self.batch_size = 8
        self.learning_rate = 5e-5
        # 权重衰减系数
        self.weight_decay = 0
        # 学习率预热的系数
        self.warmup_ratio = 0.06
        self.p_embedding_num = 6
        self.max_label_len = 2
        self.epochs = 5
        self.logging_steps = 1
        self.valid_steps = 10
        self.save_dir = r'checkpoints'


if __name__ == '__main__':
    pc = ProjectConfig()
    print(pc.verbalizer)