Browse Source

!6576 TFRecordDataset add notes

Merge pull request !6576 from heleiwang/mod_tfrecord_api
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
ea9870bf05
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

@@ -3737,6 +3737,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