Skip to content

Commit 8015851

Browse files
committed
bugfix on config
1 parent 26e8b00 commit 8015851

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,7 @@ def train_model(config: ModelConfig):
193193
train_dataloader, val_dataloader, tokenizer_src, tokenizer_tgt = get_ds(config)
194194
model = get_model(config, tokenizer_src.get_vocab_size(), tokenizer_tgt.get_vocab_size()).to(device)
195195

196-
optimizer = torch.optim.Adam(model.parameters(), lr=config['lr'], eps=1e-9)
196+
optimizer = torch.optim.Adam(model.parameters(), lr=config.lr, eps=1e-9)
197197

198198
# By default, load the latest checkpoint
199199
initial_epoch = 0

0 commit comments

Comments
 (0)