Commit 99a216bf authored by Yuxin Wu's avatar Yuxin Wu

small fix on docs

parent c3f5307e
......@@ -165,7 +165,7 @@ class ScheduledHyperParamSetter(HyperParamSetter):
def __init__(self, param, schedule, interp=None):
"""
:param schedule: [(epoch1, val1), (epoch2, val2), (epoch3, val3), ...]
(ep, val) means set the param to `val` after the `ep`th epoch.
(ep, val) means set the param to "val" after the `ep`th epoch.
If epoch == 0, the value is set before training.
:param interp: None: no interpolation. 'linear': linear interpolation
"""
......
......@@ -27,8 +27,8 @@ for _, module_name, _ in walk_packages(
class LinearWrap(object):
""" A simple wrapper to easily create linear graph,
for layers with one input&output, or tf function with one input&output
""" A simple wrapper to easily create "linear" graph,
consisting of layers / symbolic functions with only one input & output.
"""
class TFModuleFunc(object):
......
......@@ -14,8 +14,8 @@ def ConcatWith(x, dim, tensor):
A wrapper around `tf.concat` to support `LinearWrap`
:param x: the input tensor
:param dim: the dimension along which to concatenate
:param tensor: a tensor or list of tensor to concatenate with x. x will be
at the beginning
:param tensor: a tensor or list of tensor to concatenate with x.
x will be at the beginning
:return: tf.concat(dim, [x] + [tensor])
"""
if type(tensor) != list:
......
......@@ -10,7 +10,7 @@ from datetime import datetime
from six.moves import input
import sys
__all__ = []
__all__ = ['set_logger_dir', 'disable_logger', 'auto_set_dir', 'warn_dependency']
class _MyFormatter(logging.Formatter):
def format(self, record):
......
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