Suchinthana commited on
Commit
d36caf8
1 Parent(s): 8b34e2f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -1
README.md CHANGED
@@ -14,6 +14,7 @@ datasets:
14
  ### Fine tuned MT5 base model with Sinhala Wikipedia Dataset (Experimentally continues training)
15
 
16
  This model is fine tuned with articles from Sinhala Wikipedia for article generation. Used around 10,000 articles for training and fine tuned arround 100 times.
 
17
 
18
  ### How to use
19
 
@@ -44,4 +45,6 @@ tokenizer = AutoTokenizer.from_pretrained('google/mt5-base')
44
  from transformers import pipeline
45
  generator = pipeline(model='Suchinthana/MT5-Sinhala-Wikigen-Experimental', tokenizer=tokenizer)
46
  generator("writeWiki: මානව ආහාර", do_sample=True, max_length=180)
47
- ```
 
 
 
14
  ### Fine tuned MT5 base model with Sinhala Wikipedia Dataset (Experimentally continues training)
15
 
16
  This model is fine tuned with articles from Sinhala Wikipedia for article generation. Used around 10,000 articles for training and fine tuned arround 100 times.
17
+
18
 
19
  ### How to use
20
 
 
45
  from transformers import pipeline
46
  generator = pipeline(model='Suchinthana/MT5-Sinhala-Wikigen-Experimental', tokenizer=tokenizer)
47
  generator("writeWiki: මානව ආහාර", do_sample=True, max_length=180)
48
+ ```
49
+
50
+ If this model shows overfitting an early stopped version available at [non experimental repo.](https://huggingface.co/Suchinthana/MT-5-Sinhala-Wikigen)