change max count

train_main.py

@@ -16,7 +16,7 @@ def process_dataset_with_augmentation(df, is_training=False):
     # 1. Calculate Statistics for Balancing (Only needed for training)
     if is_training:
         class_counts = df['target'].value_counts().to_dict()
-        max_count = max(class_counts.values())
+        max_count = 500
         print(f"  [Augmentation] Balancing classes to match majority count: {max_count}")
 
     total = len(df)