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
b0fb5f3f
Commit
b0fb5f3f
authored
Sep 27, 2017
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
update docs
parent
99b9c39c
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
3 additions
and
1 deletion
+3
-1
README.md
README.md
+1
-0
tensorpack/graph_builder/model_desc.py
tensorpack/graph_builder/model_desc.py
+2
-1
No files found.
README.md
View file @
b0fb5f3f
...
...
@@ -8,6 +8,7 @@ A neural net training interface based on TensorFlow.
See some
[
examples
](
examples
)
to learn about the framework:
### Vision:
+
(Coming soon) Multi-GPU Faster-RCNN training on COCO.
+
[
Multi-GPU training of ResNet on ImageNet
](
examples/ResNet
)
+
[
Generative Adversarial Network(GAN) variants
](
examples/GAN
)
, including DCGAN, InfoGAN, Conditional GAN, WGAN, BEGAN, DiscoGAN, Image to Image, CycleGAN.
+
[
DoReFa-Net: train binary / low-bitwidth CNN on ImageNet
](
examples/DoReFa-Net
)
...
...
tensorpack/graph_builder/model_desc.py
View file @
b0fb5f3f
...
...
@@ -134,7 +134,7 @@ class ModelDesc(ModelDescBase):
def
get_cost
(
self
):
"""
Return the cost tensor in the graph.
Used by some of the tensorpack :class:`Trainer` which assumes single-cost models
.
It will be called by :func:`get_cost_and_grad` by default
.
You can ignore this method (or just use :class:`ModelDescBase`)
if you use your own trainer with more than one cost.
...
...
@@ -175,6 +175,7 @@ class ModelDesc(ModelDescBase):
def
get_cost_and_grad
(
self
):
"""
Compute gradients with ``self.get_optimizer()`` on ``self.get_cost()``.
This method will be used by all the existing tensorpack trainers.
Returns:
cost (tf.Tensor): the cost tensor returned by ``self.get_cost()``.
...
...
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