summaryrefslogtreecommitdiff
path: root/train_colab.ipynb
AgeCommit message (Collapse)Author
12 hoursFix total_mem → total_memory in Colab GPU checkhaoyuren
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
12 hoursFix invalid notebook cell schema (markdown with execution_count)haoyuren
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
12 hoursBatched game collection for ~7x training speeduphaoyuren
- 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>
13 hoursUpdate README and Colab notebook for current rules and featureshaoyuren
- README: document current game rules (SWAP inheritance, free draw, Q removal) - README: add versus.py usage, training features (warmup, CSV log, CPU/GPU) - Colab: update training commands, add log display, fix eval device Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
22 hoursAdd tqdm progress bar, fix Colab usernamehaoyuren
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
23 hoursAdd Colab GPU training notebookhaoyuren
Clone → train on GPU → download or push model back to GitHub. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>