Sudoku-RWKV

Sudoku-RWKV简介

Sudoku-RWKV是一个专门为解决数独谜题而设计的RWKV模型。该模型基于RWKV-v6架构，经过训练以高效地解决各类数独难题。现有的sudoku_rwkv_20241120.pth模型是基于200万数独样本（约39.2亿个token）进行训练的，具备约1270万参数，词汇量为133，包含8层和320维的架构。该模型在设计上进行了简单的改进，以提升其性能。

使用场景

1. 数独谜题求解：用户可以通过运行demo.py或minimum_inference.py脚本来解决数独谜题，适用于爱好者及学习者。
2. 训练数据生成：可以运行generate_sudoku_data.py来生成训练数据，适合研究人员和开发者在此基础上进行进一步的模型训练或优化。

通过该模型，用户能够有效解决不同难度的数独谜题，模型在各个难度级别的测试结果均显示出良好的求解准确性。这使得Sudoku-RWKV不仅适合个人娱乐，也能用于教育和机器学习研究等多个领域。