未验证 提交 81b66db9 编写于 作者: X Xin Pan 提交者: GitHub

Merge pull request #14773 from chengduoZH/fix_io_and_data_feeder_doc

Fix io and data feeder doc
...@@ -258,10 +258,13 @@ class DataFeeder(object): ...@@ -258,10 +258,13 @@ class DataFeeder(object):
multiple mini-batches. Each mini-batch will be feed on each device. multiple mini-batches. Each mini-batch will be feed on each device.
Args: Args:
reader(fun): the input data. reader(function): the reader is the function which can generate data.
multi_devices(bool): the number of places. Default None. multi_devices(bool): whether to use multiple devices or not.
num_places(int): the number of places. Default None. num_places(int): if the multi_devices is True, you can specify the number
drop_last(bool): the number of places. Default None. of GPU to use, if 'num_places' is None, the function will use all the
GPU of the current machine. Default None.
drop_last(bool): whether to drop the last batch if the
size of the last batch is less than batch_size. Default True.
Returns: Returns:
dict: the result of conversion. dict: the result of conversion.
......
...@@ -145,7 +145,7 @@ def save_vars(executor, ...@@ -145,7 +145,7 @@ def save_vars(executor,
prog = fluid.default_main_program() prog = fluid.default_main_program()
fluid.io.save_vars(executor=exe, dirname=path, main_program=prog, fluid.io.save_vars(executor=exe, dirname=path, main_program=prog,
vars=None) vars=None, predicate = name_has_fc)
# All variables in `main_program` whose name includes "fc" will be saved. # All variables in `main_program` whose name includes "fc" will be saved.
# And variables are going to be saved separately. # And variables are going to be saved separately.
...@@ -369,7 +369,7 @@ def load_vars(executor, ...@@ -369,7 +369,7 @@ def load_vars(executor,
prog = fluid.default_main_program() prog = fluid.default_main_program()
fluid.io.load_vars(executor=exe, dirname=path, main_program=prog, fluid.io.load_vars(executor=exe, dirname=path, main_program=prog,
vars=None) vars=None, predicate=name_has_fc)
# All variables in `main_program` whose name includes "fc" will be loaded. # All variables in `main_program` whose name includes "fc" will be loaded.
# And all the variables are supposed to have been saved in differnet files. # And all the variables are supposed to have been saved in differnet files.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册