| 55 | |
| 56 | @dataclass |
| 57 | class CategoryConfig: |
| 58 | categories: List[str] |
| 59 | input_dim: int = 768 |
| 60 | inner_dim: int = 128 |
| 61 | dropout: float = 0.1 |
| 62 | activation = nn.ReLU |
| 63 | embedding_dim: int = 8 |
| 64 | file_embedding_dim: int = 32 |
| 65 | |
| 66 | |
| 67 | class CommitClassifier(nn.Module): |