Spaces:
Sleeping
Sleeping
File size: 186 Bytes
542680b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
from dataclasses import dataclass
@dataclass
class ModelConfig:
bert_output_size = 312
embedding_size = 128
@dataclass
class TrainConfig:
epochs = 12
batch_size = 16
|