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

Update README.md

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