Merge branch 'master' of https://git0.fmf.uni-lj.si/klanecek/AIM

Problem 5/training.py

@@ -11,7 +11,7 @@ import time                         # For time tracking during training
 #                           DATA PREPARATION                               #
 ############################################################################
 # Define the main folder containing all processed data
-main_path_to_data = r"C:\Users\Klanecek\Desktop\processed"  # Change this path as needed
+main_path_to_data = "/data/PSUF_naloge/5-naloga/processed/"  # Change this path as needed
 
 # Get file lists from the subdirectories 'mild' and 'severe'
 files_mild = os.listdir(os.path.join(main_path_to_data, "mild"))
@@ -60,7 +60,6 @@ train_dataset = DataReader(main_path_to_data, train_info)
 valid_dataset = DataReader(main_path_to_data, valid_info)
 
 # Create DataLoader objects for training and validation
-# Batch size of 16 for training; 10 for validation (assuming one patient per batch if needed)
 train_loader = DataLoader(train_dataset, batch_size=16, shuffle=True,)
 valid_loader = DataLoader(valid_dataset, batch_size=10, shuffle=False)