muzanxdem commited on
Commit
797bd73
1 Parent(s): 5509203

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -10,7 +10,8 @@ Model = distilbert-base-uncased
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,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()```