Commit ·
c7c168c
1
Parent(s): 82406fe
- example_uses.md +1 -1
example_uses.md
CHANGED
|
@@ -17,5 +17,5 @@ python ./distill_bert_to_lstm.py --bert_model "vinai/phobert-base-v2" --bert_mod
|
|
| 17 |
|
| 18 |
- Inference with distilled LSTM model (test_data.csv is test dataset with 4 classes like ag_news)
|
| 19 |
```
|
| 20 |
-
python ./inference_lstm.py --model_path "./output/distilled_lstm_model.pth" --
|
| 21 |
```
|
|
|
|
| 17 |
|
| 18 |
- Inference with distilled LSTM model (test_data.csv is test dataset with 4 classes like ag_news)
|
| 19 |
```
|
| 20 |
+
python ./inference_lstm.py --model_path "./output/distilled_lstm_model.pth" --bert_tokenizer "vinai/phobert-base-v2" --num_classes 4 --label_column "individual" "groups" "religion/creed" "race/ethnicity" "politics" --text_column "content" --data_path "./datasets/test.csv" --inference_batch_limit 10
|
| 21 |
```
|