gpt

2023-06-08 11:18:44 +02:00 · 2023-06-08 11:18:44 +02:00 · 06b9a6297d
commit 06b9a6297d
parent 8671f27d19 54babc424b
1 changed files with 10 additions and 7 deletions
--- a/gonito.yaml
+++ b/gonito.yaml
@ -1,11 +1,14 @@
 description: trigram model
 tags:
  - neural-network
-  - trigram
+  - ngram
 params:
-  epochs: 1
-  learning-rate: 0.0001
-  vocab_size: 38000
-  embed_size: 300
-  hidden_size: 128
-  batch_size: 800
+  - vocab_size = 15000
+  - embedding_dim = 200
+  - hidden_dim = 500
+  - batch_size = 4000
+  - lr = 0.0001
+  - l_size = 25
+  - r_size = 25
+  - context_dim = 50
+