diff --git a/gonito.yaml b/gonito.yaml index 7288059..0f98fbb 100644 --- a/gonito.yaml +++ b/gonito.yaml @@ -1,7 +1,6 @@ -description: gpt2 -tags: - - neural-network - - gpt2 + - embeddings params: - top: 20 - \ No newline at end of file + epochs: 1 + learning-rate: 0.0001 + vocab_size: 5000 + embed_size: 300 \ No newline at end of file