Downloads · 30 days
0
HaileyStorm/chess-mamba-vs-xformer
chess-mamba-vs-xformer is a machine learning model from HaileyStorm. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
For an explanation of this project and the models trained for it, please see the Report.
Downloads · 30 days
0
Access
Public
Updated May 19, 2024
Repo size
92.6 GB
Likes
2
Public
Click a slice to open those files.
.pt49.7 GB · 68%
From the Hugging Face model README
For an explanation of this project and the models trained for it, please see the Report.
The root folder contains scripts for dataset preprocessing.
chess-mamba-vs-xformer contains the training scripts.
Config files, used to set model configuration and training hyperameters, are in chess-mamba-vs-xformer/config.
Model checkpoints are in chess-mamba-vs-xformer/out. The last checkpoint for completed models (e.g. Mamba and Transformer 50M) are .../anneal/anneal_complete.pt.
chess-gpt-eval has the scripts for model evaluation - playings games against Stockfish or lc0 chess engines. The logs folder contains raw evaluation metrics.
chess-gpt-eval-contrastive likewise has the scripts for model evaluation, but modified for training and evaluation of contrastive activation and linear probes. The logs folder again contains raw evaluation metrics.