File size: 511 Bytes
e4147e9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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