NanoGPT_charToken / config /eval_gpt2_xl.py
RashiAgarwal's picture
Upload 6 files
4015e59
raw
history blame
213 Bytes
# evaluate the base gpt2
# n_layer=48, n_head=25, n_embd=1600
# 1558M parameters
batch_size = 8
eval_iters = 500 # use more iterations to get good estimate
eval_only = True
wandb_log = False
init_from = 'gpt2-xl'