Skip to main content

Ctrl+K

Orbax documentation

Checkpointing

Announcements
Checkpointing with Orbax
API Overview
Using the Refactored CheckpointManager API
Checkpointing PyTrees of Arrays
Optimized Checkpointing with Tensorstore
Customizing Checkpointing Logic
Transformations
Preemption Tolerance
Asynchronous Checkpointing
orbax.checkpoint API Reference

Exporting

Exporting with Orbax
orbax.export API Reference

Additional Information

Contributors

.rst

Checkpoint logging

Contents

AbstractLogger
CompositeLogger
StandardLogger
CloudLogger
CloudLoggerOptions
StepStatistics

Checkpoint logging#

Defines exported symbols for package orbax.checkpoint.logging.

AbstractLogger#

CompositeLogger#

StandardLogger#

CloudLogger#

CloudLoggerOptions#

StepStatistics#

previous

Metadata Utilities

next

Exporting with Orbax

Contents

AbstractLogger
CompositeLogger
StandardLogger
CloudLogger
CloudLoggerOptions
StepStatistics

By Orbax Contributors

© Copyright 2024, Google.