tff.learning.LoopImplementation
Stay organized with collections
Save and categorize content based on your preferences.
An enum to specify which implementation of the training loop to use.
Attributes |
DATASET_ITERATOR
|
A training loop that uses Dataset iterator ops. This is
required when running on hosts with multiple GPUs to effectively leverage
all GPUs on the host.
|
DATASET_REDUCE
|
A training loop that uses the DatasetReduce op. This is
required when running on hosts with TPUs to allow the MLIR Bridge to
compile the reduction function to XLA HLO for TPUs.
|
Class Variables |
DATASET_ITERATOR
|
<LoopImplementation.DATASET_ITERATOR: 1>
|
DATASET_REDUCE
|
<LoopImplementation.DATASET_REDUCE: 2>
|
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.
Last updated 2024-08-23 UTC.
[{
"type": "thumb-down",
"id": "missingTheInformationINeed",
"label":"Missing the information I need"
},{
"type": "thumb-down",
"id": "tooComplicatedTooManySteps",
"label":"Too complicated / too many steps"
},{
"type": "thumb-down",
"id": "outOfDate",
"label":"Out of date"
},{
"type": "thumb-down",
"id": "samplesCodeIssue",
"label":"Samples / code issue"
},{
"type": "thumb-down",
"id": "otherDown",
"label":"Other"
}]
[{
"type": "thumb-up",
"id": "easyToUnderstand",
"label":"Easy to understand"
},{
"type": "thumb-up",
"id": "solvedMyProblem",
"label":"Solved my problem"
},{
"type": "thumb-up",
"id": "otherUp",
"label":"Other"
}]
{"lastModified": "Last updated 2024-08-23 UTC."}