diff --git a/README.md b/README.md index 8ddd586..d6f87cd 100644 --- a/README.md +++ b/README.md @@ -113,9 +113,9 @@ Performs classification on Binary SST by default. Output classification probabilities are saved to a `.npy` file ``` -python3 run_classifier.py --load_model ama_sst.pt # classify Binary SST -python3 run_classifier.py --load_model ama_sst_16.pt --fp16 # run classification in fp16 -python3 run_classifier.py --load_model ama_sst.pt --text-key --data # classify your own dataset +python3 run_classifier.py --load ama_sst.pt # classify Binary SST +python3 run_classifier.py --load ama_sst_16.pt --fp16 # run classification in fp16 +python3 run_classifier.py --load ama_sst.pt --text-key --data # classify your own dataset ``` See [here](./script_docs/arguments.md#running-a-classifier-arguments) for more documentation.