Browse Source

TFRecordDataset add notes

tags/v1.0.0
heleiwang 5 years ago
parent
commit
13ed9f99b3
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      mindspore/dataset/engine/datasets.py

+ 4
- 0
mindspore/dataset/engine/datasets.py View File

@@ -3725,6 +3725,10 @@ class TFRecordDataset(SourceDataset):
"""
Get the number of batches in an epoch.

Note:
Because the TFRecord format does not save metadata, all files need to be traversed to obtain
the total amount of data. Therefore, this api is slow.

Args:
estimate (bool, optional): Fast estimation of the dataset size instead of a full scan.



Loading…
Cancel
Save