orbit.utils.make_distributed_dataset
Stay organized with collections
Save and categorize content based on your preferences.
A utility function to help create a tf.distribute.DistributedDataset
.
orbit.utils.make_distributed_dataset(
strategy, dataset_or_fn, *args, **kwargs
)
Args |
strategy
|
An instance of tf.distribute.Strategy .
|
dataset_or_fn
|
A instance of tf.data.Dataset , or a "dataset function"
returning a tf.data.Dataset . If it is a function, it may optionally have
an argument named input_context which will be passed a
tf.distribute.InputContext instance.
|
*args
|
Any positional arguments to pass through to dataset_or_fn .
|
**kwargs
|
Any keyword arguments to pass through to dataset_or_fn , except
that the input_options keyword is used to specify a
tf.distribute.InputOptions for making the distributed dataset.
|
Returns |
A distributed Dataset.
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2024-07-19 UTC.
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2024-07-19 UTC."],[],[]]