bugfix on config

Author: hkproj

train.py

@@ -193,7 +193,7 @@ def train_model(config: ModelConfig):
     train_dataloader, val_dataloader, tokenizer_src, tokenizer_tgt = get_ds(config)
     model = get_model(config, tokenizer_src.get_vocab_size(), tokenizer_tgt.get_vocab_size()).to(device)
 
-    optimizer = torch.optim.Adam(model.parameters(), lr=config['lr'], eps=1e-9)
+    optimizer = torch.optim.Adam(model.parameters(), lr=config.lr, eps=1e-9)
 
     # By default, load the latest checkpoint
     initial_epoch = 0