rlightning.models.modules.value_head¶

class rlightning.models.modules.value_head.ValueHead(input_dim: int, hidden_sizes=(512, 128), output_dim: int = 1, activation: str = 'gelu', bias_last: bool = False)[source]¶

Bases: torch.nn.Module