Joshua Lochner commited on
Commit
90d506c
1 Parent(s): 1286fe5

Fix classifier train command

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -118,5 +118,5 @@ If you wish to finetune an original transformer model, use one of the supported
118
  ### Classifier
119
  The classifier is used to add probabilities to the category predictions. Train the classifier using:
120
  ```bash
121
- python src/train.py --do_train_classifier --skip_train_transformer
122
  ```
 
118
  ### Classifier
119
  The classifier is used to add probabilities to the category predictions. Train the classifier using:
120
  ```bash
121
+ python src/train.py --train_classifier --skip_train_transformer
122
  ```