from transformers import PretrainedConfig | |
class ArabiCharModelConfig(PretrainedConfig): | |
def __init__( | |
self, | |
num_classes=28, | |
conv1_channels=32, | |
conv2_channels=64, | |
fc1_units=128, | |
dropout_prob=0.4, | |
**kwargs | |
): | |
super().__init__(**kwargs) | |
self.num_classes = num_classes | |
self.conv1_channels = conv1_channels | |
self.conv2_channels = conv2_channels | |
self.fc1_units = fc1_units | |
self.dropout_prob = dropout_prob |