| import os |
| import torch |
|
|
| class ArgsConfig: |
| def __init__(self) -> None: |
| self.batch_size = 192 |
| self.embedding_size = 480 |
| self.epochs = 50 |
| self.kflod = 5 |
| self.max_len = 40 |
| self.lr = 1.5e-3 |
| self.weight_decay = 0 |
| self.is_autocast = False |
| self.info_bottleneck = False |
| self.dropout = 0.6 |
| self.IB_beta = 1e-3 |
| self.model_name = 'DeepPD_C' |
| self.exp_nums = 0.0 |
| self.aa_dict = 'esm' |
| self.info = f"" |
| |
| |
| |
| |
| |
|
|
| |
| |
| |
| self.ems_path = './DeepPD/ESM2/esm2_t12_35M_UR50D.pt' |
| self.esm_layer_idx = 12 |
| |
| self.random_seed = 2023 |
| self.num_classes = 21 |
| self.split_size = 0.8 |
| self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu") |
| |
