Data

---

download data

mkdir -p /data/kata/npzs/train
cd /data/kata/npzs/train
wget https://katagoarchive.org/kata1/trainingdata/2025-09-01npzs.tgz
tar -xzf 2025-09-01npzs.tgz

preprocess data

# modify ./data/npz_preprocess.py 'day_start and day_end' with '2025-09-01'
python ./data/npz_preprocess.py

Training

---

run with 8 gpus

mkdir -p /data/kata/code/
mv KataFormer /data/kata/code/
cd /data/kata/code/KataFormer/
pip install -r ./requirements.txt

chmod +x run.sh
./run.sh

TODO

---

• Use opset 23 when TensorRT adds rms_norm support.
• mxfp8 & nvfp4