Skip to content

Commit 8b3e4b2

Browse files
authored
Remove unused params for hparam script (rasbt#710)
1 parent 190c66b commit 8b3e4b2

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

ch05/05_bonus_hparam_tuning/hparam_search.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -64,8 +64,7 @@ def evaluate_model(model, train_loader, val_loader, device, eval_iter):
6464

6565

6666
def train_model(model, train_loader, val_loader, optimizer, device,
67-
n_epochs, eval_freq, eval_iter,
68-
encoded_start_context, tokenizer, warmup_iters=10,
67+
n_epochs, eval_iter, warmup_iters=10,
6968
initial_lr=3e-05, min_lr=1e-6):
7069
global_step = 0
7170

@@ -192,9 +191,7 @@ def train_model(model, train_loader, val_loader, optimizer, device,
192191
train_loss, val_loss = train_model(
193192
model, train_loader, val_loader, optimizer, device,
194193
n_epochs=HPARAM_CONFIG["n_epochs"],
195-
eval_freq=5, eval_iter=1,
196-
encoded_start_context=encoded_tensor,
197-
tokenizer=tokenizer,
194+
eval_iter=1,
198195
warmup_iters=HPARAM_CONFIG["warmup_iters"],
199196
initial_lr=HPARAM_CONFIG["initial_lr"],
200197
min_lr=HPARAM_CONFIG["min_lr"]

0 commit comments

Comments
 (0)