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
f221d7f3
Commit
f221d7f3
authored
Dec 10, 2018
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
update docs
parent
82aa4b2b
Changes
5
Show whitespace changes
Inline
Side-by-side
Showing
5 changed files
with
7 additions
and
6 deletions
+7
-6
README.md
README.md
+1
-1
examples/DoReFa-Net/README.md
examples/DoReFa-Net/README.md
+1
-1
examples/FasterRCNN/README.md
examples/FasterRCNN/README.md
+1
-1
examples/ResNet/README.md
examples/ResNet/README.md
+1
-1
tensorpack/callbacks/inference_runner.py
tensorpack/callbacks/inference_runner.py
+3
-2
No files found.
README.md
View file @
f221d7f3
...
...
@@ -64,7 +64,7 @@ Dependencies:
+
Python 2.7 or 3.3+. Python 2.7 is supported until
[
it retires in 2020
](
https://pythonclock.org/
)
.
+
Python bindings for OpenCV (Optional, but required by a lot of features)
+
TensorFlow
>=
1.3. (If you only want to use
`tensorpack.dataflow`
alone as a data processing library, TensorFlow is not needed)
+
TensorFlow
≥
1.3. (If you only want to use
`tensorpack.dataflow`
alone as a data processing library, TensorFlow is not needed)
```
pip install --upgrade git+https://github.com/tensorpack/tensorpack.git
# or add `--user` to install to user's local directories
...
...
examples/DoReFa-Net/README.md
View file @
f221d7f3
...
...
@@ -41,7 +41,7 @@ In this implementation, quantized operations are all performed through `tf.float
## Use
+
Install TensorFlow
>=
1.7, tensorpack and scipy.
+
Install TensorFlow
≥
1.7, tensorpack and scipy.
+
Look at the docstring in
`*-dorefa.py`
to see detailed usage and performance.
...
...
examples/FasterRCNN/README.md
View file @
f221d7f3
...
...
@@ -14,7 +14,7 @@ with the support of:
## Dependencies
+
Python 3.3+; OpenCV.
+
TensorFlow
>=
1.6 (1.4 or 1.5 can run but may crash due to a TF bug);
+
TensorFlow
≥
1.6 (1.4 or 1.5 can run but may crash due to a TF bug);
+
pycocotools:
`pip install 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'`
+
Pre-trained
[
ImageNet ResNet model
](
http://models.tensorpack.com/FasterRCNN/
)
from tensorpack model zoo.
...
...
examples/ResNet/README.md
View file @
f221d7f3
...
...
@@ -9,7 +9,7 @@ __Training__ code of three variants of ResNet on ImageNet:
The training follows the __exact__ recipe used by the
[
Training ImageNet in 1 Hour paper
](
https://arxiv.org/abs/1706.02677
)
and gets the same performance.
Distributed training
code & results can be found at
[
tensorpack/benchmarks
](
https://github.com/tensorpack/benchmarks/tree/master/ResNet-Horovod
)
.
__Distributed training__
code & results can be found at
[
tensorpack/benchmarks
](
https://github.com/tensorpack/benchmarks/tree/master/ResNet-Horovod
)
.
This recipe has better performance than most open source implementations.
In fact, many papers that claim to "improve" ResNet by .5% only compete with a lower
...
...
tensorpack/callbacks/inference_runner.py
View file @
f221d7f3
...
...
@@ -202,9 +202,10 @@ class DataParallelInferenceRunner(InferenceRunnerBase):
tower_name (str): the name scope of the tower to build. Need to set a
different one if multiple InferenceRunner are used.
tower_func (tfutils.TowerFuncWrapper or None): the tower function to be used to build the graph.
By defaults to call `trainer.tower_func` under a `training=False` TowerContext,
The tower function will be called under a `training=False` TowerContext.
The default is `trainer.tower_func`,
but you can change it to a different tower function
if you need to inference with several different
graph
s.
if you need to inference with several different
model
s.
"""
if
isinstance
(
gpus
,
int
):
gpus
=
list
(
range
(
gpus
))
...
...
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