File size: 784 Bytes
d814291
 
 
 
 
 
 
 
 
 
 
 
 
fe1f842
d814291
 
 
 
 
 
 
 
 
 
 
 
fe1f842
d814291
 
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
29
30
"""Adversarial self-play training helpers."""

from osint_env.training.config import (
    GeneratorRewardWeights,
    KimiGRPOPhaseConfig,
    LoraTuningConfig,
    SelfPlayTrainingConfig,
    SwarmV2Config,
    SwarmV2SharedContextConfig,
    SwarmV2SwarmConfig,
    SwarmV2ValidationConfig,
    load_self_play_config,
)
from osint_env.training.hf_jobs import launch_hf_self_play_job
from osint_env.training.self_play import run_adversarial_self_play

__all__ = [
    "GeneratorRewardWeights",
    "KimiGRPOPhaseConfig",
    "LoraTuningConfig",
    "SelfPlayTrainingConfig",
    "SwarmV2Config",
    "SwarmV2SharedContextConfig",
    "SwarmV2SwarmConfig",
    "SwarmV2ValidationConfig",
    "load_self_play_config",
    "launch_hf_self_play_job",
    "run_adversarial_self_play",
]