Update README.md

README.md

@@ -89,7 +89,19 @@ python3 prepare.py
 ```
 If all data has loaded, you can start the training:
 ```bash
-python3 train.py
+python train.py \
+    --n_layer=10 \
+    --n_head=8 \
+    --n_embd=256 \
+    --block_size=512 \
+    --batch_size=32 \
+    --gradient_accumulation_steps=4 \
+    --max_iters=5000 \
+    --eval_interval=100 \
+    --learning_rate=5e-4 \
+    --compile=False \
+    --dtype='float16' \
+    --device='cuda'
 ```
 Then, you'll have to wait until iteration 5000 is reached (will log something like `iter 5000: loss 4.2044, time 50601.67ms, mfu 2.23%`).