facepalm

lucidrains · Oct 2, 2023 · 983573b · 983573b
1 parent 78fde6f
commit 983573b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'speculative-decoding',
   packages = find_packages(exclude=[]),
-  version = '0.0.15',
+  version = '0.0.16',
   license='MIT',
   description = 'Speculative Decoding',
   author = 'Phil Wang',

diff --git a/train_prophet.py b/train_prophet.py
@@ -113,12 +113,12 @@ def __len__(self):
 
 # optimizer
 
-optim = Adam(model.parameters(), lr = LEARNING_RATE)
+optim = Adam(model_and_prophet.parameters(), lr = LEARNING_RATE)
 
 # training
 
 for i in tqdm.tqdm(range(NUM_BATCHES), mininterval = 10.0, desc = "training"):
-    model.train()
+    model_and_prophet.train()
 
     for _ in range(GRAD_ACCUM_EVERY):
         data = next(train_loader)