Merge pull request #22 from lrm22005/Luis

labels updated on path
lrm22005 · Apr 20, 2024 · 09dec70 · 09dec70
2 parents 650ee24 + 5f89620
commit 09dec70
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/GP_Original_checkpoint.py b/GP_Original_checkpoint.py
@@ -279,7 +279,7 @@ def load_checkpoint(self, loader_name):
     data_path, labels_path, saving_path = get_data_paths(data_format, is_linux=is_linux, is_hpc=is_hpc)
 
     # Define batch size for loading data
-    batch_size = 512
+    batch_size = 1024
     # Load the training, validation, and test data
     train_loader = load_data_split_batched(data_path, labels_path, clinical_trial_train, batch_size, standardize=True, data_format='csv', read_all_labels=False, drop_last=True)
     val_loader = load_data_split_batched(data_path, labels_path, clinical_trial_test, batch_size, standardize=True, data_format='csv', read_all_labels=False, drop_last=True)

diff --git a/__pycache__/GP_original_data.cpython-312.pyc b/__pycache__/GP_original_data.cpython-312.pyc