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

update docs

parent 04d60559
......@@ -26,7 +26,8 @@ class ModelSaver(Callback):
keep_recent=None, keep_freq=None):
"""
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()``.
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