Learn more »
Push, build, and install
RubyGems
npm packages
Python packages
Maven artifacts
PHP packages
Go Modules
Bower components
Debian packages
RPM packages
NuGet packages
/ distributed / checkpoint / __init__.py
from .metadata import (
TensorStorageMetadata,
BytesStorageMetadata,
ChunkStorageMetadata,
Metadata,
)
from .state_dict_loader import load_state_dict
from .state_dict_saver import save_state_dict
from .storage import StorageReader, StorageWriter
from .filesystem import FileSystemReader, FileSystemWriter
from .api import CheckpointException
from .planner import (
SavePlanner,
LoadPlanner,
SavePlan,
LoadPlan,
ReadItem,
WriteItem,
)
from .default_planner import DefaultSavePlanner, DefaultLoadPlanner