Update train.py

train.py

@@ -21,10 +21,10 @@ model = BertForSequenceClassification.from_pretrained('bert-base-uncased', num_l
 training_args = TrainingArguments(
     output_dir="./results",
     evaluation_strategy="epoch",
-    learning_rate=2e-5,
+    learning_rate=2e-3,
     per_device_train_batch_size=8,
     per_device_eval_batch_size=8,
-    num_train_epochs=3,
+    num_train_epochs=10,
     weight_decay=0.01,
 )