diff --git a/nobrainer/dataset.py b/nobrainer/dataset.py index f2d6e8bb..fe57d357 100644 --- a/nobrainer/dataset.py +++ b/nobrainer/dataset.py @@ -147,7 +147,7 @@ def from_tfrecords( label_mapping=label_mapping, num_parallel_calls=num_parallel_calls ) - ds_obj.filter_zero_volumes() + # ds_obj.filter_zero_volumes() # TODO automatically determine batch size ds_obj.batch(1)