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 |