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

small fix on docs

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