Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
S
seminar-breakout
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Issues
0
Issues
0
List
Boards
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Analytics
Analytics
CI / CD
Repository
Value Stream
Wiki
Wiki
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Shashank Suhas
seminar-breakout
Commits
2aefaf78
Commit
2aefaf78
authored
Sep 09, 2018
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
fix setup.cfg (fix #889)
parent
8250786f
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
5 additions
and
4 deletions
+5
-4
docs/tutorial/save-load.md
docs/tutorial/save-load.md
+3
-3
setup.cfg
setup.cfg
+2
-1
No files found.
docs/tutorial/save-load.md
View file @
2aefaf78
# 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,
...
...
setup.cfg
View file @
2aefaf78
...
@@ -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
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment