Commit a8e47470 authored by vfdev's avatar vfdev Committed by Yuxin Wu

Update tower.py (#719)

parent 881338a0
...@@ -146,7 +146,7 @@ class SingleCostTrainer(TowerTrainer): ...@@ -146,7 +146,7 @@ class SingleCostTrainer(TowerTrainer):
Args: Args:
inputs_desc ([InputDesc]): inputs_desc ([InputDesc]):
input (InputSource): input (InputSource):
get_cost_fn ([tf.Tensor] -> tf.Tensor): callable, takes some input tenosrs and return a cost tensor. get_cost_fn ([tf.Tensor] -> tf.Tensor): callable, takes some input tensors and return a cost tensor.
get_opt_fn (-> tf.train.Optimizer): callable which returns an get_opt_fn (-> tf.train.Optimizer): callable which returns an
optimizer. Will only be called once. optimizer. Will only be called once.
......
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