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
9c42c84f
Commit
9c42c84f
authored
Jan 01, 2017
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
sort varnames in logging
parent
4e7977f5
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
8 additions
and
6 deletions
+8
-6
README.md
README.md
+5
-3
tensorpack/tfutils/sessinit.py
tensorpack/tfutils/sessinit.py
+3
-3
No files found.
README.md
View file @
9c42c84f
...
...
@@ -5,7 +5,7 @@ See some [examples](examples) to learn about the framework:
### Vision:
+
[
DoReFa-Net: train binary / low-bitwidth CNN on ImageNet
](
examples/DoReFa-Net
)
+
[
Train ResNet on ImageNet
/Cifar10/
SVHN
](
examples/ResNet
)
+
[
Train ResNet on ImageNet
/ Cifar10 /
SVHN
](
examples/ResNet
)
+
[
InceptionV3 on ImageNet
](
examples/Inception/inceptionv3.py
)
+
[
Fully-convolutional Network for Holistically-Nested Edge Detection(HED)
](
examples/HED
)
+
[
Spatial Transformer Networks on MNIST addition
](
examples/SpatialTransformer
)
...
...
@@ -58,10 +58,12 @@ The components are designed to be independent. You can use Model or DataFlow in
+
other requirements:
```
pip install --user -r requirements.txt
pip install --user -r opt-requirements.txt (some optional dependencies, you can install later if needed)
pip install --user -r opt-requirements.txt
#
(some optional dependencies, you can install later if needed)
```
+
Enable
`import tensorpack`
(or use
`greadlink`
from
`coreutils`
brew package if you're on OSX)
:
+
Enable
`import tensorpack`
:
```
export PYTHONPATH=$PYTHONPATH:`readlink -f path/to/tensorpack`
```
(or use
`greadlink`
from
`coreutils`
brew package if you're on OSX)
+
Use tcmalloc if running with large data
tensorpack/tfutils/sessinit.py
View file @
9c42c84f
...
...
@@ -141,7 +141,7 @@ class SaverRestore(SessionInit):
logger
.
warn
(
"Variable {} in the graph not found in checkpoint!"
.
format
(
v
.
op
.
name
))
if
len
(
chkpt_vars_used
)
<
len
(
vars_available
):
unused
=
vars_available
-
chkpt_vars_used
for
name
in
unused
:
for
name
in
sorted
(
unused
)
:
if
not
is_training_name
(
name
):
logger
.
warn
(
"Variable {} in checkpoint not found in the graph!"
.
format
(
name
))
return
var_dict
...
...
@@ -167,10 +167,10 @@ class ParamRestore(SessionInit):
logger
.
info
(
"Params to restore: {}"
.
format
(
', '
.
join
(
map
(
str
,
intersect
))))
for
k
in
variable_names
-
param_names
:
for
k
in
sorted
(
variable_names
-
param_names
)
:
if
not
is_training_name
(
k
):
logger
.
warn
(
"Variable {} in the graph not found in the dict!"
.
format
(
k
))
for
k
in
param_names
-
variable_names
:
for
k
in
sorted
(
param_names
-
variable_names
)
:
logger
.
warn
(
"Variable {} in the dict not found in the graph!"
.
format
(
k
))
upd
=
SessionUpdate
(
sess
,
...
...
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