File size: 579 Bytes
1e99f6a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
from transformers import PretrainedConfig
from typing import List
class ThaiLightWeightEncoderConfig(PretrainedConfig):
def __init__(
self,
input_embedding_dim: int = 300,
final_embedding_dim: int = 512,
dropout: float = 0.2,
word_vector_model_name: str = "thai2fit_wv",
**kwargs,
):
self.input_embedding_dim = input_embedding_dim
self.final_embedding_dim = final_embedding_dim
self.word_vector_model_name = word_vector_model_name
self.dropout = dropout
super().__init__(**kwargs) |