import torch

class Config:
    def __init__(self):
        # 1- 设备
        self.device = ("cuda" if torch.cuda.is_available() else "cpu")

        # 2- 原始文件路径
        self.dev_datapath = "data/dev.txt"
        self.class_datapath = "data/class.txt"

        # 3- 数据加载器参数
        self.batch_size = 64  # 64条新闻
        self.max_length = 32  # 句子中词的个数最多是32个词

        # 4- Bert预训练模型的路径
        self.bert_path = r"D:\soft\PretrainedModel\bert-base-chinese"

        # 5- 目标值文件解析
        self.classname_list = [line.strip() for line in open(self.class_datapath, mode="r", encoding="UTF-8")]
        self.classname_len = len(self.classname_list)

        # 6- 剪枝前模型的保存路径
        self.before_prune_path = "save_model/before_prune_model.pkl"

        # 7- 剪枝后模型的保存路径
        self.after_prune_path = "save_model/after_prune_model.pkl"