resolved #278

neuronets · Mar 12, 2024 · c67bd9e · c67bd9e
1 parent ed0d609
commit c67bd9e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/nobrainer/dataset.py b/nobrainer/dataset.py
@@ -121,6 +121,9 @@ def from_tfrecords(
         )
         block_length = len([0 for _ in first_shard])
 
+        if not n_volumes:
+            n_volumes = block_length * len(files)
+
         dataset = dataset.interleave(
             map_func=lambda x: tf.data.TFRecordDataset(
                 x, compression_type=compression_type