Skip to content

Commit e2afddf

Browse files
holgerrothwyli
andauthored
fix debug dataset path (#772)
Co-authored-by: Wenqi Li <[email protected]>
1 parent b105849 commit e2afddf

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

federated_learning/breast_density_challenge/code/pt/learners/mammo_learner.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -646,11 +646,11 @@ def fire_event(self, event_type: str, fl_ctx: FLContext):
646646
inside_container = True
647647
if inside_container:
648648
debug_dataset_root = "/data/preprocessed"
649-
debug_datalist_prefix = "/data/dataset_blinded_phase1_"
649+
debug_datalist_prefix = "/data/dataset_blinded_"
650650
else:
651651
# assumes script is run in from repo root, e.g. using `python3 code/pt/learners/mammo_learner.py`
652652
debug_dataset_root = "./data/preprocessed"
653-
debug_datalist_prefix = "./data/dataset_blinded_phase1_"
653+
debug_datalist_prefix = "./data/dataset_blinded_"
654654

655655
print("Testing MammoLearner...")
656656
learner = MammoLearner(

0 commit comments

Comments
 (0)