| import os | |
| from typing import Dict, List, Optional, Union | |
| from transformers.configuration_utils import PretrainedConfig | |
| from transformers.utils import logging | |
| logger = logging.get_logger(__name__) | |
| class PDeepPPConfig(PretrainedConfig): | |
| model_type = "PDeepPP" | |
| def __init__( | |
| self, | |
| input_size=1280, | |
| output_size=128, | |
| num_heads=8, | |
| hidden_size=256, | |
| num_transformer_layers=4, | |
| dropout=0.3, | |
| task_type="", # 留空，依赖 convert 文件动态补充 | |
| esm_ratio=None, # 留空，依赖 convert 文件动态补充 | |
| lambda_=None, # 留空，依赖 convert 文件动态补充 | |
| **kwargs | |
| ): | |
| super().__init__(**kwargs) | |
| self.input_size = input_size | |
| self.output_size = output_size | |
| self.num_heads = num_heads | |
| self.hidden_size = hidden_size | |
| self.num_transformer_layers = num_transformer_layers | |
| self.dropout = dropout | |
| self.task_type = task_type # 默认留空 | |
| self.esm_ratio = esm_ratio # 默认留空 | |
| self.lambda_ = lambda_ # 默认留空 | |
| PDeepPPConfig.register_for_auto_class() |