saiful9379 commited on
Commit
30206d4
1 Parent(s): 371d540

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -1
README.md CHANGED
@@ -29,4 +29,14 @@ output = model.generate(
29
  )
30
  predicted_text = tokenizer.decode(output[0], skip_special_tokens=True)
31
  print(predicted_text)
32
- ```
 
 
 
 
 
 
 
 
 
 
 
29
  )
30
  predicted_text = tokenizer.decode(output[0], skip_special_tokens=True)
31
  print(predicted_text)
32
+ ```
33
+
34
+ Model configuration,
35
+ Here is the basic configuration of Bangla GPT2 Model,
36
+ ```
37
+ vocab_size = 50000
38
+ block_size = 200
39
+ learning_rate=3e-5
40
+ num_epoch = 100
41
+ batch_size = 12
42
+ buffer_size = 1000