Remove prints in split_train_val

Continvvm · Jan 12, 2022 · d490d28 · d490d28
1 parent acce31e
commit d490d28
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/continuum/tasks/utils.py b/continuum/tasks/utils.py
@@ -62,9 +62,6 @@ def split_train_val(dataset: BaseTaskSet, val_split: float = 0.1) -> Tuple[BaseT
     train_indexes = indexes[int(val_split * len(indexes)):]
     val_indexes = indexes[:int(val_split * len(indexes))]
 
-    print(train_indexes)
-    print(val_indexes)
-
     if dataset.data_type != TaskType.H5:
         x_train, y_train, t_train = dataset.get_raw_samples(train_indexes)
         x_val, y_val, t_val = dataset.get_raw_samples(val_indexes)