tuned params

Author: Ubuntu

src/bart_reddit_lora/train.py

@@ -49,7 +49,7 @@ class CustomTrainingArgs(Seq2SeqTrainingArguments):
     num_train_epochs: int = 6
     per_device_train_batch_size: int = 8
     per_device_eval_batch_size: int = 16
-    learning_rate: float = 7e-5
+    learning_rate: float = 6e-5
     lr_scheduler_type: str = "cosine"
     warmup_ratio: float = 0.1
     max_grad_norm: float = 0.5