Restorer

Inheritance Diagram

Inheritance diagram of ashpy.restorers.restorer.Restorer

class ashpy.restorers.restorer.Restorer(logdir='log', ckpts_dir='ckpts', expect_partial=True)[source]

Bases: object

Restorer provide a way to restore objects from tf.train.Checkpoint.

Can be standalone.

Methods

__init__([logdir, ckpts_dir, expect_partial]) Initialize the Restorer.
get_global_step() Return the restored global_step.
get_steps_per_epoch() Return the restored global_step.
restore_callback(callback, callback_ckpt_id) Return the restored callbacks.
restore_object(placeholder, object_ckpt_id) Restore a placeholder from a checkpoint using the specified id.

Attributes

checkpoint_map Get the map of the ids in the checkpoint.
__init__(logdir='log', ckpts_dir='ckpts', expect_partial=True)[source]

Initialize the Restorer.

Parameters:
  • logdir (str) – Path to the directory with the logs.
  • ckpts_dir (str) – Name of the directory with the checkpoints to restore.
  • expect_partial (bool) – Whether to expect partial restoring or not. Default to true. For more information see the docs for tf.train.Checkpoint.restore().
Return type:

None

_restore_checkpoint(checkpoint, partial=True)[source]

Restore or initialize the persistence layer (checkpoint).

checkpoint_map

Get the map of the ids in the checkpoint.

Return type:Dict[str, str]
get_global_step()[source]

Return the restored global_step.

Return type:Variable
get_steps_per_epoch()[source]

Return the restored global_step.

Return type:Variable
restore_callback(callback, callback_ckpt_id)[source]

Return the restored callbacks.

Return type:List[Callback]
restore_object(placeholder, object_ckpt_id)[source]

Restore a placeholder from a checkpoint using the specified id.

Warning

When restoring a tf.keras.Model object from checkpoint assure that the model has been correctly built and instantiated by firstly calling it on some sample inputs. In the case of a model built with either the Sequential or Functional API an exception will be raised; for a model built with the Chainer API it will fail silently, restoration will be “successful” but no values will actually be restored since there are no valid placeholder as the model has not be built yet.

TODO: Args TODO: Example