Update README.md

README.md

@@ -64,7 +64,7 @@ python3 pretrain.py --dataset_path couplet_dataset.pt \
                     --learning_rate 5e-4 --batch_size 64 \
                     --embedding word_pos --remove_embedding_layernorm \
                     --encoder transformer --mask causal --layernorm_positioning pre \
-                    --target lm --tie_weight
+                    --target lm --tie_weights
 ```
 
 Finally, we convert the pre-trained model into Huggingface's format: