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