dbd41fe
1
2
3
4
5
6
7
8
9
10
11
VOCAB_SIZE = 256 BLOCK_SIZE = 64 N_LAYER = 1 N_HEAD = 4 N_EMBD = 4 BATCH_SIZE = 32 LR = 0.05 EPOCHS = 2
