Update README.md

README.md

@@ -58,7 +58,7 @@ The command used for training is the following. For more information about data
 ```python
 cd task_forward
 python train.py \
-    --model='t5' \
+    --output_dir='t5' \
     --epochs=100 \
     --lr=1e-3 \
     --batch_size=32 \