tf.contrib.eager.restore_network_checkpoint( network, save_path, map_func=None )
Restore the Network from a checkpoint.
If variables have already been created (typically when some or all of the
Network is built), they are assigned values from the checkpoint immediately,
overwriting any existing values (in graph mode the default session is used for
If there are checkpoint entries which do not correspond to any existing
variables in the
Network, these values are saved for deferred restoration;
their initial values will be the checkpointed values once they are
created. Requests for multiple deferred restorations behave the same way as
immediate restorations, in that later requests will take priority over earlier
requests relevant to the same variable.
Layers with another network, those
also have their variables restored from the checkpoint.
network: A Network object to restore.
save_path: The return value of
tfe.save_network_checkpoint, or a directory to search for a checkpoint.
map_func: A function mapping fully qualified variable names (e.g. 'my_network_1/dense_1/kernel') to names in the checkpoint. By default (if
map_func=None), the variable prefix for the network being restored (
Network.scope_name + '/', e.g. 'my_network_1/') is stripped and all other variable names (shared with other Networks) are left unchanged. Note that this is the same map_func as
tfe.save_network_checkpoint, not an inverse mapping.