From abbf749029a09390b3253e252070a4700b4e4764 Mon Sep 17 00:00:00 2001
From: Tibo De Peuter <tibo.depeuter@ugent.be>
Date: Sun, 7 Dec 2025 22:05:59 +0100
Subject: [PATCH] chore: Update README usage

---
 README.md | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index a183021..2b0b5f7 100644
--- a/README.md
+++ b/README.md
@@ -3,7 +3,9 @@
 Example usage:
 
 ```shell
-python main_cnn.py --debug train --dataset enwik9 --method optuna
+python main.py --debug train --dataset enwik9 --data-root ~/data/datasets/ml --method optuna --model transformer --model-save-path ~/data/ml-models/test-transformer.pt
+
+python benchmark.py --debug train --dataset enwik9 --data-root ~/data/datasets/ml --method optuna --model cnn --model-save-path ~/data/ml-models/test-cnn.pt
 ```
 
 ## Running locally