Update README.md
Browse files
README.md
CHANGED
@@ -17,24 +17,34 @@ from transformers import (
|
|
17 |
Text2TextGenerationPipeline,
|
18 |
)
|
19 |
|
20 |
-
|
|
|
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 |
+
|