From 41184a27f535b79cea9f7ac2ef31edd908d040a2 Mon Sep 17 00:00:00 2001 From: Andrej Karpathy Date: Mon, 2 Jan 2023 01:15:55 +0000 Subject: [PATCH] rename compile_model to compile, shroter, version 2 stragglers --- bench.py | 4 ++-- config/finetune_shakespeare.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/bench.py b/bench.py index 60db2bb..840dcc5 100644 --- a/bench.py +++ b/bench.py @@ -15,7 +15,7 @@ torch.manual_seed(1337) batch_size = 8 block_size = 1024 dtype = torch.bfloat16 -compile_model = True +compile = True # data loading init real_data = True @@ -47,7 +47,7 @@ model.to(device) optimizer = model.configure_optimizers(weight_decay=1e-2, learning_rate=1e-4, betas=(0.9, 0.95)) -if compile_model: +if compile: print("Compiling model...") model = torch.compile(model) # pytorch 2.0 diff --git a/config/finetune_shakespeare.py b/config/finetune_shakespeare.py index 69d49a6..eb6545e 100644 --- a/config/finetune_shakespeare.py +++ b/config/finetune_shakespeare.py @@ -5,7 +5,7 @@ eval_interval = 200 wandb_log = False # feel free to turn on wandb_project = 'shakespeare' wandb_run_name = 'ft-' + str(time.time()) -compile_model = False # takes too little time to finetune, not worth it +compile = False # takes too little time to finetune, not worth it # save a nice and overfit checkpoint that # will only speak Shakespeare and forgets