update

README.md

@@ -27,9 +27,7 @@ Download your tokenizer, model checkpoints, and optionally the training logs (`e
 Optionally, test model using the MLM task:
 ```sh
 pip install pya0
-python test.py \
-    --model_name_or_path ./ckpt/to/tokenizer \
-    --tokenizer_name_or_path ./ckpt/to/tokenizer
+python test.py ./ckpt/to/tokenizer ./ckpt/to/model
 ```
 > **Note**  
 > Modify the test examples in `test.txt` to play with it.

test.py

@@ -25,7 +25,7 @@ def classifier_hook(tokenizer, tokens, topk, module, inputs, outputs):
             str(tokenizer.convert_ids_to_tokens(top_cands)))
 
 
-def test(model_name_or_path, tokenizer_name_or_path, test_file='test.txt'):
+def test(tokenizer_name_or_path, model_name_or_path, test_file='test.txt'):
 
     tokenizer = AutoTokenizer.from_pretrained(tokenizer_name_or_path)
     model = AutoModelForPreTraining.from_pretrained(model_name_or_path,