Tien-THM commited on
Commit
7865af0
·
1 Parent(s): 6681371

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -5
README.md CHANGED
@@ -38,8 +38,8 @@ Use the code below to get started with the model.
38
  # Load model directly
39
  from transformers import AutoTokenizer, TFAutoModelForQuestionAnswering
40
 
41
- tokenizer = AutoTokenizer.from_pretrained("Tien-THM/QAVi")
42
- model = TFAutoModelForQuestionAnswering.from_pretrained("Tien-THM/QAVi")
43
 
44
  import numpy as np
45
 
@@ -79,9 +79,6 @@ Using 2 datasets:
79
  * Learning rate: 2e-5
80
  * Batch size: 8
81
  * Epoch: 4
82
- 'train_loss': [4.7110, 3.2650, 2.7899, 2.4633],
83
- 'val_loss': [3.6251, 3.3062, 3.2184, 3.1946],
84
- 'exact_match': [0.38, 0.42, 0.44, 0.45]
85
  #### Training Loss
86
  | Epoch | Train loss | Validation loss | Exact Match |
87
  |----------|----------|----------|----------|
 
38
  # Load model directly
39
  from transformers import AutoTokenizer, TFAutoModelForQuestionAnswering
40
 
41
+ tokenizer = AutoTokenizer.from_pretrained("Tien-THM/bert-mini-fine-tuning-squad")
42
+ model = TFAutoModelForQuestionAnswering.from_pretrained("Tien-THM/bert-mini-fine-tuning-squad")
43
 
44
  import numpy as np
45
 
 
79
  * Learning rate: 2e-5
80
  * Batch size: 8
81
  * Epoch: 4
 
 
 
82
  #### Training Loss
83
  | Epoch | Train loss | Validation loss | Exact Match |
84
  |----------|----------|----------|----------|