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 |
model = TFDistilBertForSequenceClassification.from_pretrained("distilbert-base-uncased")
|
15 |
|
16 |
trainer = TFTrainer(
|
@@ -20,4 +20,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 |
+
`with training_args.strategy.scope():
|
14 |
model = TFDistilBertForSequenceClassification.from_pretrained("distilbert-base-uncased")
|
15 |
|
16 |
trainer = TFTrainer(
|
|
|
20 |
eval_dataset=test_dataset # evaluation dataset
|
21 |
)
|
22 |
|
23 |
+
trainer.train()`
|