You need to sign in or sign up before continuing.
提交 4649f662 编写于 作者: Y yuyang18

Follow comments & polish doc

上级 9942a304
...@@ -103,6 +103,10 @@ def convert_reader_to_recordio_files( ...@@ -103,6 +103,10 @@ def convert_reader_to_recordio_files(
This API is basically same as :code:`convert_reader_to_recordio_file`, This API is basically same as :code:`convert_reader_to_recordio_file`,
instead of it will create many recordio files. Each file contains at instead of it will create many recordio files. Each file contains at
most :code:`batch_per_file` records. most :code:`batch_per_file` records.
Please reference
:ref:`api_fluid_recordio_writer_convert_reader_to_recordio_file` for more
details.
""" """
if feed_order is None: if feed_order is None:
feed_order = feeder.feed_names feed_order = feeder.feed_names
......
...@@ -83,7 +83,7 @@ class EndStepEvent(object): ...@@ -83,7 +83,7 @@ class EndStepEvent(object):
epoch_id(int): The current epoch ID. epoch_id(int): The current epoch ID.
step_id(int): The current step ID. step_id(int): The current step ID.
metrics(list): A list of fetched tensor. The order of this list is same metrics(list): A list of fetched tensor. The order of this list is same
as the :code:`train_func` returns. as the :code:`train_func` returns.
""" """
def __init__(self, epoch_id, step_id, metrics): def __init__(self, epoch_id, step_id, metrics):
...@@ -99,7 +99,7 @@ class CheckpointConfig(object): ...@@ -99,7 +99,7 @@ class CheckpointConfig(object):
Args: Args:
checkpoint_dir(str): Directory path to save check point. Default is the checkpoint_dir(str): Directory path to save check point. Default is the
current directory. current directory.
max_num_checkpoints(int): The max number of local check points. max_num_checkpoints(int): The max number of local check points.
epoch_interval(int): Every number of epoch to save check point. epoch_interval(int): Every number of epoch to save check point.
...@@ -389,11 +389,11 @@ class Trainer(object): ...@@ -389,11 +389,11 @@ class Trainer(object):
Start the train loop to train the model. Start the train loop to train the model.
Args: Args:
num_epochs: The number of epoch. An epoch will process all data in reader num_epochs(int): The number of epoch. An epoch will process all data in reader
event_handler: The event handler. A function with type (ev:Event)->void event_handler(callable): The event handler. A function with type (ev:Event)->void
reader: A reader creator object. See also reader(callable): A reader creator object. See also
:ref:`api_guide_python_reader` . :ref:`api_guide_python_reader` .
feed_order: Feeding order of reader. None will following the defining feed_order(list): Feeding order of reader. None will following the defining
order in program order in program
Returns: Returns:
...@@ -427,9 +427,10 @@ class Trainer(object): ...@@ -427,9 +427,10 @@ class Trainer(object):
def save_params(self, param_path): def save_params(self, param_path):
""" """
Save all parameters into :code:`param_path` Save all parameters into :code:`param_path`.
Args: Args:
param_path(str): The path to save parameters param_path(str): The path to save parameters.
Returns: Returns:
None None
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册