We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 26e8b00 commit 8015851Copy full SHA for 8015851
train.py
@@ -193,7 +193,7 @@ def train_model(config: ModelConfig):
193
train_dataloader, val_dataloader, tokenizer_src, tokenizer_tgt = get_ds(config)
194
model = get_model(config, tokenizer_src.get_vocab_size(), tokenizer_tgt.get_vocab_size()).to(device)
195
196
- optimizer = torch.optim.Adam(model.parameters(), lr=config['lr'], eps=1e-9)
+ optimizer = torch.optim.Adam(model.parameters(), lr=config.lr, eps=1e-9)
197
198
# By default, load the latest checkpoint
199
initial_epoch = 0
0 commit comments