remove training code

This commit is contained in:
Jai Suphavadeeprasit 2026-03-13 12:52:52 -04:00
parent 862cd3667d
commit 148a4fd5eb
6 changed files with 38 additions and 329 deletions

View file

@ -69,18 +69,6 @@ class TrainingConfig(BaseModel):
"Prevents large policy updates that could destabilize training."
),
)
distill_enabled: bool = Field(
False,
description="Enable teacher distillation loss when distill tensors are present.",
)
distill_coef: float = Field(
0.0,
description="Weight for distillation loss in total loss.",
)
distill_temperature: float = Field(
1.0,
description="Temperature applied when converting teacher top-k logprobs.",
)
# === Device & Storage ===
device: str = Field(
"cuda" if torch.cuda.is_available() else "cpu", description="Device to train on"