diff --git a/python/mxnet/metric.py b/python/mxnet/metric.py index 60d1a69d..bc51e469 100644 --- a/python/mxnet/metric.py +++ b/python/mxnet/metric.py @@ -318,7 +318,7 @@ class CustomMetric(EvalMetric): The name of the metric allow_extra_outputs : bool If true, the prediction outputs can have extra outputs. - This is useful in RNN, where the states are also produced + This is useful in RNN, where the states are also produced in outputs for forwarding. """ def __init__(self, feval, name=None, allow_extra_outputs=False): @@ -362,7 +362,7 @@ def np(numpy_feval, name=None, allow_extra_outputs=False): The name of the metric. allow_extra_outputs : bool If true, the prediction outputs can have extra outputs. - This is useful in RNN, where the states are also produced + This is useful in RNN, where the states are also produced in outputs for forwarding. """ def feval(label, pred): diff --git a/python/mxnet/optimizer.py b/python/mxnet/optimizer.py index 11f7d230..01263f64 100644 --- a/python/mxnet/optimizer.py +++ b/python/mxnet/optimizer.py @@ -268,7 +268,7 @@ def create_state(self, index, weight): return None else: return zeros(weight.shape, weight.context, dtype=weight.dtype) - + def update(self, index, weight, grad, state): """Update the parameters.