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 |
```
|