update train script

train.py

@@ -9,6 +9,7 @@ import numpy as np
 import torch
 from collections import Counter
 
+
 # Load the dataset from Hugging Face Hub
 dataset = load_dataset("Luigi/dinercall-intent")
 
@@ -91,12 +92,12 @@ training_args = TrainingArguments(
     learning_rate=2e-5,
     per_device_train_batch_size=16,
     per_device_eval_batch_size=16,
-    num_train_epochs=20,
+    num_train_epochs=100,
     weight_decay=0.01,
     logging_dir='./logs',
     logging_steps=10,
     save_total_limit=2,
-    push_to_hub=False,# 👈 This will push your model after training
+    push_to_hub=True,# 👈 This will push your model after training
     hub_model_id="Luigi/albert-tiny-chinese-dinercall-intent",  # optional, use if you want a custom name
     hub_private_repo=False  # optional, if you want a private repo
 )