refactor(hyfi): update verbose parameter in dataset saving to false, …

…add num_heads and num_tails parameters to dataset loading and sampling
entelecheia · Jul 26, 2023 · c0549e4 · c0549e4
1 parent 74b1661
commit c0549e4
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/src/hyfi/conf/pipe/dataset_load_from_disk.yaml b/src/hyfi/conf/pipe/dataset_load_from_disk.yaml
@@ -4,5 +4,7 @@ defaults:
 run: hyfi.pipe.datasets.load_dataset_from_disk
 run_with:
   dataset_path:
+  num_heads: 1
+  num_tails: 1
   verbose: ${..verbose}
 use_pipe_obj: false
diff --git a/src/hyfi/conf/pipe/dataset_sample.yaml b/src/hyfi/conf/pipe/dataset_sample.yaml
@@ -6,6 +6,8 @@ run_with:
   num_samples: 100
   randomize: true
   random_seed: 123
+  num_heads: 1
+  num_tails: 1
   verbose: ${..verbose}
 use_pipe_obj: true
 return_pipe_obj: false
diff --git a/src/hyfi/conf/pipe/dataset_save_to_disk.yaml b/src/hyfi/conf/pipe/dataset_save_to_disk.yaml
@@ -4,3 +4,4 @@ defaults:
 run: hyfi.pipe.datasets.save_dataset_to_disk
 run_with:
   dataset_path:
+  verbose: false