VOCAB_SIZE = 256
BLOCK_SIZE = 64

N_LAYER = 1
N_HEAD = 4
N_EMBD = 4

BATCH_SIZE = 32
LR = 0.05
EPOCHS = 2