diff --git a/gonito.yaml b/gonito.yaml index 1c62d3b..ad57d2b 100644 --- a/gonito.yaml +++ b/gonito.yaml @@ -1,11 +1,6 @@ -description: bilstm, left context +description: gpt-2, left context, zero-shot tags: - - bilstm + - gpt2 + - just-inference params: - epochs: 1 - vocab-size: 5000 - batch-size: 64 - embed-size: 50 - lstm-size: 300 - topk: 200 - learning-rate: 1.0e-3 \ No newline at end of file + topk: 150 \ No newline at end of file