@@ -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)
