Upload README.md with huggingface_hub

README.md

@@ -77,9 +77,9 @@ training_args = TrainingArguments(
     logging_dir=f"../logs/{model_dir}/",
     logging_strategy='epoch',
     num_train_epochs=10,
-    per_device_train_batch_size=args.batch,
-    per_device_eval_batch_size=args.batch,
-    learning_rate=args.lr,
+    per_device_train_batch_size=8,
+    per_device_eval_batch_size=8,
+    learning_rate=5e-06,
     seed=42,
     save_strategy='epoch',
     evaluation_strategy='epoch',