Commit 2aefaf78 authored by Yuxin Wu's avatar Yuxin Wu

fix setup.cfg (fix #889)

parent 8250786f
# Save and Load models # Save and Load models
## Work with TF Checkpoint ## Inspect a TF Checkpoint
The `ModelSaver` callback saves the model to the directory defined by `logger.get_logger_dir()`, The `ModelSaver` callback saves the model to the directory defined by `logger.get_logger_dir()`,
in TensorFlow checkpoint format. in TensorFlow checkpoint format.
...@@ -21,9 +21,9 @@ demos how to print all variables and their shapes in a checkpoint. ...@@ -21,9 +21,9 @@ demos how to print all variables and their shapes in a checkpoint.
It takes a metagraph file (which is also saved by `ModelSaver`) and only saves variables that the model needs at inference time. It takes a metagraph file (which is also saved by `ModelSaver`) and only saves variables that the model needs at inference time.
It can dump the model to a `var-name: value` dict saved in npz format. It can dump the model to a `var-name: value` dict saved in npz format.
## Load a Model ## Load a Model to a Session
Model loading (in either training or testing) is through the `session_init` interface. Model loading (in either training or inference) is through the `session_init` interface.
Currently there are two ways a session can be restored: Currently there are two ways a session can be restored:
[session_init=SaverRestore(...)](../modules/tfutils.html#tensorpack.tfutils.sessinit.SaverRestore) [session_init=SaverRestore(...)](../modules/tfutils.html#tensorpack.tfutils.sessinit.SaverRestore)
which restores a TF checkpoint, which restores a TF checkpoint,
......
...@@ -7,7 +7,8 @@ license = Apache ...@@ -7,7 +7,8 @@ license = Apache
[options] [options]
zip_safe = False # dataset and __init__ use file zip_safe = False # dataset and __init__ use file
packages = find: # will call find_packages() # will call find_packages()
packages = find:
[wheel] [wheel]
universal = 1 universal = 1
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment