experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Minor fix to snapshot documentation to add snapshot path as positional argument

Browse Source 
PiperOrigin-RevId: 314260887
Change-Id: If3a771a0ba7c6454e9022477614f7bea7f80f404
This commit is contained in:
Frank Chen 2020-06-01 21:03:11 -07:00 committed by TensorFlower Gardener
parent a3f393bc95
commit 9fbf5e5d50
1 changed files with 2 additions and 2 deletions

4
tensorflow/python/data/experimental/ops/snapshot.py
View File

@ -284,7 +284,7 @@ def snapshot(path, compression="AUTO", reader_func=None, shard_func=None):
```python
dataset = ...
dataset = dataset.enumerate()
dataset = dataset.apply(tf.data.experimental.snapshot(
dataset = dataset.apply(tf.data.experimental.snapshot("/path/to/snapshot/dir",
shard_func=lambda x, y: x % NUM_SHARDS, ...))
dataset = dataset.map(lambda x, y: y)
```
@ -308,7 +308,7 @@ def snapshot(path, compression="AUTO", reader_func=None, shard_func=None):
# read datasets in parallel and interleave their elements
return datasets.interleave(lambda x: x, num_parallel_calls=AUTOTUNE)
dataset = dataset.apply(tf.data.experimental.snapshot(
dataset = dataset.apply(tf.data.experimental.snapshot("/path/to/snapshot/dir",
reader_func=user_reader_func))
```