Update train_main.py

train_main.py

@@ -59,6 +59,9 @@ def process_dataset_with_augmentation(df, is_training=False):
 
 
 def main():
+    """
+    This is the starting point of the training.
+    It produces -between others- plots and a model saved in serialized format (classes.pkl) """
     # 1. Load Data (Metadata only)
     df, classes = data.load_metadata(limit=None)  # Adjust limit as needed