Changes for submission-strategyA

local_evaluation.py

@@ -34,29 +34,29 @@ DATASET_SHUFFLE_SEED = 1022022
 
 # Instantiate Training Dataset
 training_dataset = ZEWDPCBaseDataset(
-    images_dir="./data/debug/images",
-    labels_path="./data/debug/labels.csv",
+    images_dir="./data/training/images",
+    labels_path="./data/training/labels.csv",
     shuffle_seed=DATASET_SHUFFLE_SEED,
 )
 # Instantiate Unlabelled Dataset
 unlabelled_dataset = ZEWDPCProtectedDataset(
-    images_dir="./data/debug/images",
-    labels_path="./data/debug/labels.csv",
-    purchase_budget=50,  # Configurable Parameter
+    images_dir="./data/unlabelled/images",
+    labels_path="./data/unlabelled/labels.csv",
+    purchase_budget=300,  # Configurable Parameter
     shuffle_seed=DATASET_SHUFFLE_SEED,
 )
 # Instantiate Validation Dataset
 val_dataset = ZEWDPCBaseDataset(
-    images_dir="./data/debug/images",
-    labels_path="./data/debug/labels.csv",
+    images_dir="./data/validation/images",
+    labels_path="./data/validation/labels.csv",
     drop_labels=True,
     shuffle_seed=DATASET_SHUFFLE_SEED,
 )
 # A second instantiation of the validation test with the labels present
 #       - helpful later, when computing the scores.
 val_dataset_gt = ZEWDPCBaseDataset(
-    images_dir="./data/debug/images",
-    labels_path="./data/debug/labels.csv",
+    images_dir="./data/validation/images",
+    labels_path="./data/validation/labels.csv",
     drop_labels=False,
     shuffle_seed=DATASET_SHUFFLE_SEED,
 )
@@ -75,8 +75,8 @@ checkpoint_folder_path = tempfile.TemporaryDirectory().name
 ####################################################################################
 time_started = time.time()
 
-PURCHASE_BUDGET = 50
-COMPUTE_BUDGET = 200
+PURCHASE_BUDGET = 300
+COMPUTE_BUDGET = 900
 
 # ####################################################################################
 ####################################################################################