grain.checkpoint.CheckpointHandler

class grain.checkpoint.CheckpointHandler

Orbax CheckpointHandler for PyGrain iterators.

__init__()

Methods

__init__()
close()
finalize(directory)
metadata(directory)
restore(directory[, item, args])	Restores the given iterator from the checkpoint in directory.
save(directory[, item, args])	Saves the given iterator to the checkpoint in directory.
structure(directory)
typestr()