Skip to content

Commit

Permalink
Add exploration file for mnist dataset
Browse files Browse the repository at this point in the history
  • Loading branch information
klei22 committed Apr 15, 2024
1 parent f982d76 commit 7bc7110
Showing 1 changed file with 28 additions and 0 deletions.
28 changes: 28 additions & 0 deletions explorations/mnist.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
[
{
"max_iters": ["30000"],
"eval_iters": ["200"],
"eval_interval": ["250"],
"log_interval": ["10"],
"n_layer": ["8"],
"n_kv_group": ["4"],
"n_head": ["8"],
"n_embd": ["384"],
"block_size": ["256"],
"shared_mlp_size" : ["1"],
"shared_mlp_sym" : [true],
"shared_attn_size" : ["2"],
"shared_attn_sym" : [true],
"device": ["cuda"],
"dataset": ["mnist"],
"compile": [true],
"use_post_ln": [false],
"softmax_variant_attn": ["softmax"],
"use_abs_pos_embeddings": [false],
"use_rotary_embeddings": [false],
"use_fire_embeddings": [true],
"shared_fire_embeddings": [true],
"tensorboard_run_name": ["mnist_ascii"]
}
]

0 comments on commit 7bc7110

Please sign in to comment.