MicroRWKV / config /eval_gpt2.py
AnshulRanjan2004's picture
Uploading the Model
c50fe14 verified
raw
history blame contribute delete
208 Bytes
# evaluate the base gpt2
# n_layer=12, n_head=12, n_embd=768
# 124M parameters
batch_size = 8
eval_iters = 500 # use more iterations to get good estimate
eval_only = True
wandb_log = False
init_from = 'gpt2'