File size: 448 Bytes
386302b | 1 2 3 4 5 6 7 8 9 10 11 12 13 | """Hugging Face configuration for TinyLunarPolicy (32-param linear LunarLander policy)."""
from transformers import PretrainedConfig
class TinyLunarConfig(PretrainedConfig):
model_type = "tiny-lunar-policy"
def __init__(self, obs_dim: int = 8, n_actions: int = 4, bias: bool = False, **kwargs):
self.obs_dim = int(obs_dim)
self.n_actions = int(n_actions)
self.bias = bool(bias)
super().__init__(**kwargs)
|