Commit b0fb5f3f authored by Yuxin Wu's avatar Yuxin Wu

update docs

parent 99b9c39c
......@@ -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)
......
......@@ -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()``.
......
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