Spaces:

blitz1809
/

sre-arena

Sleeping

App Files Files Community

sre-arena / training

49.8 kB

Ctrl+K

Ctrl+K

3 contributors

History: 17 commits

blitz1809's picture

fix: switch to bf16 for HF Jobs (newer transformers + bnb prefers bf16, no GradScaler needed)

e43130f 3 months ago

__init__.py

49 Bytes
Phase 6a: prompts and action parser with unit tests 3 months ago
action_parser.py

2.94 kB
Phase 6a: prompts and action parser with unit tests 3 months ago
alternating_loop.py

4.58 kB
Phase 7a: PEFT checkpoint loading for cross-generation training 3 months ago
dataset_builder.py

2.54 kB
feat(6c): add build_attacker_rollout_dataset to dataset_builder 3 months ago
opponent_loader.py

6.91 kB
feat(7a): OpponentModel — frozen PEFT checkpoint loader with blind generation 3 months ago
plots.py

3.49 kB
feat(6c): plots.py with reward curve + cross-gen matrix stub + 4 tests 3 months ago
prompts.py

3.26 kB
Phase 6a: prompts and action parser with unit tests 3 months ago
reward_function.py

6.08 kB
Phase 7a: PEFT checkpoint loading for cross-generation training 3 months ago
scripted_attacker.py

1.79 kB
Phase 6b: defender training script with scripted attacker 3 months ago
scripted_defender.py

791 Bytes
feat(6c): scripted_defender baseline + 5 tests 3 months ago
train_attacker.py

8.24 kB
fix: switch to bf16 for HF Jobs (newer transformers + bnb prefers bf16, no GradScaler needed) 3 months ago
train_defender.py

9.17 kB
fix: switch to bf16 for HF Jobs (newer transformers + bnb prefers bf16, no GradScaler needed) 3 months ago