| 140 | |
| 141 | @config_class |
| 142 | class Config(BaseClassificationHead.Config): |
| 143 | inner_head: Required[InstantiableConfig] = REQUIRED # Output layer. |
| 144 | transform: InstantiableConfig = NonLinear.default_config() # Output transform. |
| 145 | ignored_target_id: int = -100 # Value of target that should be ignored. |
| 146 | |
| 147 | def __init__(self, cfg: Config, *, parent: Optional[Module]): |
| 148 | super().__init__(cfg, parent=parent) |
nothing calls this directly
no test coverage detected