gonito.yaml

2023-06-08 03:00:10 +02:00 · 2023-06-08 03:00:10 +02:00 · 2987fbdbe9
parent 4a69a01029
commit 2987fbdbe9
1 changed files with 2 additions and 12 deletions
--- a/gonito.yaml
+++ b/gonito.yaml
@ -1,18 +1,8 @@
-description: nn with bagging
+description: gpt2-small
 tags:
  - neural-network
-  - ngram
+  - gpt2
 params:
-  epochs: 1
-  learning-rate: 0.0001
-  vocab_size: 25000
-  embed_size: 300
-  bagging_left_ctx: 25
-  ngram_left_ctx: 7
-  bagging_right_ctx: 25
-  ngram_right_ctx: 3
-  hidden_size: 150
-  batch_size: 4000
 unwanted-params:
  - model-file
  - vocab-file