Aminrhmni commited on
Commit
0310dda
1 Parent(s): b1d6579

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -1
README.md CHANGED
@@ -17,24 +17,34 @@ from transformers import (
17
  Text2TextGenerationPipeline,
18
  )
19
 
20
- path = r".\fine-tuned_mt5"
 
21
  pipe = Text2TextGenerationPipeline(
22
  model=MT5ForConditionalGeneration.from_pretrained(path),
23
  tokenizer=T5Tokenizer.from_pretrained(path),
24
  )
25
 
 
26
  sentence = "ویراویراست یک نرم افزار ویرایش متن ساختاری و معنایی زبان فارسی است چیزی شبیه به گرامرلی در زبان انگلیسی"
 
27
  #res = pipe(sentence, max_length=100, num_beams=4)
 
28
  res = pipe(sentence, max_length=100)
 
29
  print(res[0]['generated_text'])
30
 
31
 
32
  -----------------------
33
  n_epochs = 4
 
34
  train_batch_size = 8
 
35
  eval_batch_size = 4
 
36
  lr = 5e-4
37
 
38
  Training_loss=0.00550
 
39
  Validation_loss=0.052046
40
 
 
 
17
  Text2TextGenerationPipeline,
18
  )
19
 
20
+
21
+ path = ""
22
  pipe = Text2TextGenerationPipeline(
23
  model=MT5ForConditionalGeneration.from_pretrained(path),
24
  tokenizer=T5Tokenizer.from_pretrained(path),
25
  )
26
 
27
+
28
  sentence = "ویراویراست یک نرم افزار ویرایش متن ساختاری و معنایی زبان فارسی است چیزی شبیه به گرامرلی در زبان انگلیسی"
29
+
30
  #res = pipe(sentence, max_length=100, num_beams=4)
31
+
32
  res = pipe(sentence, max_length=100)
33
+
34
  print(res[0]['generated_text'])
35
 
36
 
37
  -----------------------
38
  n_epochs = 4
39
+
40
  train_batch_size = 8
41
+
42
  eval_batch_size = 4
43
+
44
  lr = 5e-4
45
 
46
  Training_loss=0.00550
47
+
48
  Validation_loss=0.052046
49
 
50
+