diff options
| author | haoyuren <13851610112@163.com> | 2026-02-22 11:56:48 -0600 |
|---|---|---|
| committer | haoyuren <13851610112@163.com> | 2026-02-22 11:56:48 -0600 |
| commit | 8392bdfc10f92e61303e39bb356522ee491ce97c (patch) | |
| tree | 91359c42773bf9d5fb8f1d76d393743ff4a55387 /blazing_env.py | |
| parent | 6f7034fabbfbe27197765f335bdcc64ec8c8c85f (diff) | |
Batched game collection for ~7x training speedup
- collect_games_batch(): run N games in parallel with single batched forward pass per step
- evaluate_vs_greedy_batch(): batched evaluation replacing sequential eval
- Add --collect_batch CLI arg for configurable parallel game count
- Use torch.inference_mode() for faster collection
- Update Colab notebook: GPU info, --collect_batch, log download cell
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
Diffstat (limited to 'blazing_env.py')
0 files changed, 0 insertions, 0 deletions