diff --git a/gonito.yaml b/gonito.yaml index cd0b65b..7460600 100644 --- 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