File size: 596 Bytes
04b46cd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 | """AntiAtropos Training Module.
RL loss functions and training loop for LLM-based SRE agents.
Works with the simulator in pure-Python mode (no AWS/GPU needed for validation).
"""
from .losses import (
compute_returns,
compute_gae,
reinforce_loss,
reinforce_baseline_loss,
grpo_loss,
rloo_loss,
)
from .trainer import SRETrainer, TrainingConfig, EpisodeCollector
__all__ = [
"compute_returns",
"compute_gae",
"reinforce_loss",
"reinforce_baseline_loss",
"grpo_loss",
"rloo_loss",
"SRETrainer",
"TrainingConfig",
"EpisodeCollector",
]
|