Spaces:
Running
Running
Joshua Lochner
commited on
Commit
•
90d506c
1
Parent(s):
1286fe5
Fix classifier train command
Browse files
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 --
|
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 |
```
|