From 2bcdeb6362f2991da354bbd660674c151474aa0e Mon Sep 17 00:00:00 2001
From: BinarySandia04 <aranseraroig@gmail.com>
Date: Fri, 27 Mar 2026 19:10:18 +0100
Subject: [PATCH] Fixed training of three models

---
 README.md    | 19 +++++++++++++++++++
 hf_upload.py | 38 ++++++++++++++++++++++++++++++++++++++
 train.sh     | 23 +++++++++++++++++++++++
 3 files changed, 80 insertions(+)
 create mode 100644 hf_upload.py
 create mode 100755 train.sh

diff --git a/README.md b/README.md
index 81b4559..f105cbd 100644
--- a/README.md
+++ b/README.md
@@ -1,5 +1,24 @@
+---
+license: mit
+library_name: pytorch
+tags:
+    - reinforcement-learning
+    - game-ai
+    - tic-tac-toe
+    - pytorch
+---
+
 # Ultimate Tic Tac Toe Deep Learning Bot
 
+Model for playing Ultimate Tic Tac Toe
+
+## Available checkpoints
+
+- `checkpoints/medium.pth`: medium-difficulty checkpoint.
+- `checkpoints/hard.pth`: hard-difficulty checkpoint.
+
+## Usage
+
 **Usage**
 Run `python run.py --help` for help.
 
diff --git a/hf_upload.py b/hf_upload.py
new file mode 100644
index 0000000..c39ae09
--- /dev/null
+++ b/hf_upload.py
@@ -0,0 +1,38 @@
+from pathlib import Path
+
+from huggingface_hub import HfApi, create_repo
+
+
+REPO_ID = "Syndria98/ultimate-tic-tac-toe-ai"
+
+FILES_TO_UPLOAD = {
+    "model.py": "model.py",
+    "game.py": "game.py",
+    "run.py": "run.py",
+    "README.md": "README.md",
+    "easy.pth": "models/easy.pth",
+    "medium.pth": "models/medium.pth",
+    "hard.pth": "models/hard.pth",
+}
+
+
+def main():
+    create_repo(repo_id=REPO_ID, repo_type="model", exist_ok=True)
+    api = HfApi()
+
+    for local_path, repo_path in FILES_TO_UPLOAD.items():
+        path = Path(local_path)
+        if not path.exists():
+            raise FileNotFoundError(f"Missing file: {local_path}")
+
+        api.upload_file(
+            path_or_fileobj=str(path),
+            path_in_repo=repo_path,
+            repo_id=REPO_ID,
+            repo_type="model",
+        )
+        print(f"Uploaded {local_path} -> {repo_path}")
+
+
+if __name__ == "__main__":
+    main()
diff --git a/train.sh b/train.sh
new file mode 100755
index 0000000..cc33b4e
--- /dev/null
+++ b/train.sh
@@ -0,0 +1,23 @@
+#!/bin/bash
+
+set -euo pipefail
+
+# Override with DEVICE=... ./train.sh if you want to force a specific device.
+DEVICE="${DEVICE:-$(python -c 'import torch; print("cuda" if torch.cuda.is_available() else "cpu")')}"
+OUTPUT_CHECKPOINT="${OUTPUT_CHECKPOINT:-easy.pth}"
+
+# Example heavier runs:
+DEVICE=cuda python run.py train --checkpoint hard.pth --arena-compare-games 4 --arena-compare-simulations 4 --num-simulations 25
+DEVICE=cuda python run.py train --checkpoint medium.pth --arena-compare-games 3 --arena-compare-simulations 4 --num-simulations 20 --epochs 5 --num-eps 15 --num-iters 30
+python run.py train \
+  --checkpoint "$OUTPUT_CHECKPOINT" \
+  --arena-compare-games 0 \
+  --arena-compare-simulations 2 \
+  --device "$DEVICE" \
+  --num-simulations 20 \
+  --epochs 3 \
+  --num-eps 10 \
+  --num-iters 15 \
+  "$@"
+
+echo "Training finished!"