Update README.md
Browse files
README.md
CHANGED
@@ -10,7 +10,7 @@ Model = distilbert-base-uncased
|
|
10 |
|
11 |
Fine tuned the model from our own custom dataset
|
12 |
|
13 |
-
```
|
14 |
with training_args.strategy.scope():
|
15 |
model = TFDistilBertForSequenceClassification.from_pretrained("distilbert-base-uncased")
|
16 |
|
@@ -21,4 +21,5 @@ trainer = TFTrainer(
|
|
21 |
eval_dataset=test_dataset # evaluation dataset
|
22 |
)
|
23 |
|
24 |
-
trainer.train()
|
|
|
|
10 |
|
11 |
Fine tuned the model from our own custom dataset
|
12 |
|
13 |
+
```
|
14 |
with training_args.strategy.scope():
|
15 |
model = TFDistilBertForSequenceClassification.from_pretrained("distilbert-base-uncased")
|
16 |
|
|
|
21 |
eval_dataset=test_dataset # evaluation dataset
|
22 |
)
|
23 |
|
24 |
+
trainer.train()
|
25 |
+
```
|