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