official-stockfish/nnue-pytorch
Stockfish NNUE (Chess evaluation) trainer in Pytorch
Thành tựu nổi bật
Những danh hiệu mới nhất của dự án
#1 Repository Của Tuần
#1 Repository Của Tuần
#1 Repository Của Tuần
🧠 Huấn luyện NNUE Stockfish bằng PyTorch — toàn bộ pipeline trong container
Huấn luyện mạng đánh giá NNUE dùng PyTorch, hỗ trợ CUDA, ROCm, MPS, kèm công cụ tự động đánh giá mạng.
Dự án nnue-pytorch là bộ huấn luyện mạng NNUE (Efficiently Updatable Neural Network) dùng cho động cơ cờ vua Stockfish, được viết bằng PyTorch. Hỗ trợ chạy qua Docker với CUDA hoặc ROCm, hoặc chạy trực tiếp trên macOS bằng MPS. Bao gồm cả công cụ tự động chuyển đổi checkpoint (.ckpt) sang định dạng .nnue và chạy trận đấu để xếp hạng mạng tốt nhất.
✨ Điểm nổi bật
- Docker hóa toàn bộ: hỗ trợ CUDA 12.x (NVIDIA) hoặc ROCm 6.4.3 (AMD), không cần cài đặt môi trường phức tạp.
- Hỗ trợ Apple Silicon (MPS): chạy trên macOS bằng conda hoặc micromamba, tận dụng tăng tốc phần cứng M-series.
- Tự động chuyển đổi định dạng: chuyển các file
.ckptthành.nnueđể dùng trực tiếp với Stockfish. - Chạy trận đấu tự động: dùng
c-chess-clivàordođể đánh giá và xếp hạng mạng, hỗ trợ chạy đồng thời với huấn luyện. - Giám sát thư mục: script
run_games.pyliên tục theo dõi thư mục checkpoint mới và xử lý tự động. - Ghi log TensorBoard: theo dõi quá trình huấn luyện qua giao diện web tại cổng 6006.
💡 Vì sao đáng chú ý
Không chỉ là bộ huấn luyện đơn thuần, nnue-pytorch cung cấp một pipeline hoàn chỉnh từ huấn luyện → đánh giá → chuyển đổi định dạng → chạy trận đấu. Điều này giúp người dùng dễ dàng so sánh các phiên bản mạng và tự động tìm ra mạng tốt nhất để tích hợp vào Stockfish.
#NNUE #Stockfish #PyTorch #DeepLearning #Chess #TiniX
Similar Projects
Discover other similar projects and AI models.
Implement a ChatGPT-like LLM in PyTorch from scratch, step by step
🧑🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠
🧑🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠
Ultralytics YOLO27, YOLO26, YOLO11, YOLOv8 — object detection, instance segmentation, semantic segmentation, image classification, pose estimation, object tracking
Clone a voice in 5 seconds to generate arbitrary speech in real-time
Ultralytics YOLOv5 in PyTorch for object detection, instance segmentation, classification, training, and export.
🐸💬 - a deep learning toolkit for Text-to-Speech, battle-tested in research and production
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.