Update README.md

README.md

@@ -1,5 +1,7 @@
 ---
 license: cc-by-nc-4.0
+datasets:
+- clane9/NSD-Flat
 ---
 
 # Model card for `boldgpt_small_patch10.cont`
@@ -31,3 +33,17 @@ batch["activity"] = transform(batch["activity"])
 # output: (B, N + 1, D) predicted next patches
 output, state = model(batch)
 ```
+
+## Reproducing
+
+- Training command:
+
+  ```bash
+  torchrun --standalone --nproc_per_node=4 \
+    scripts/train.py \
+    --out_dir results \
+    --model boldgpt_small_patch10 \
+    --no_cat --shuffle --epochs 1000 --bs 512 \
+    --workers 0 --amp --compile --wandb
+  ```
+- Commit: `e0b29adc8d5b3ed2f1a555d7de4754ba96a3bb3e`