chore: fixed unused lr paramaters

nguyenanht · Feb 16, 2024 · acc7edf · acc7edf
1 parent a94eb3c
commit acc7edf
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/john_toolbox/train/transformers/train.py b/john_toolbox/train/transformers/train.py
@@ -123,7 +123,6 @@ def get_dataset(config):
         src_seq_len=config["seq_len"],
         tgt_seq_len=config["seq_len"],
         emb_size=config["d_model"],
-        lr=config["lr"],
         device="auto",
         log_dir=f"/work/data/{config['experiment_name']}",
     )