Commit 68b1f125 authored by Yuxin Wu's avatar Yuxin Wu

update docs

parent 04d60559
...@@ -26,7 +26,8 @@ class ModelSaver(Callback): ...@@ -26,7 +26,8 @@ class ModelSaver(Callback):
keep_recent=None, keep_freq=None): keep_recent=None, keep_freq=None):
""" """
Args: Args:
max_to_keep, keep_checkpoint_every_n_hours(int): the same as in ``tf.train.Saver``. max_to_keep (int): the same as in ``tf.train.Saver``.
keep_checkpoint_every_n_hours (int): the same as in ``tf.train.Saver``.
checkpoint_dir (str): Defaults to ``logger.get_logger_dir()``. checkpoint_dir (str): Defaults to ``logger.get_logger_dir()``.
var_collections (str or list of str): collection of the variables (or list of collections) to save. var_collections (str or list of str): collection of the variables (or list of collections) to save.
""" """
......
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