-
Notifications
You must be signed in to change notification settings - Fork 1.3k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
added default values and one more argument "max_checkpoints" to --help #215
Conversation
…ints to prevent default deleting checkpoints if there are more than 5.
train.py
Outdated
@@ -267,7 +270,7 @@ def main(): | |||
sess.run(init) | |||
|
|||
# Saver for storing checkpoints of the model. | |||
saver = tf.train.Saver(var_list=tf.trainable_variables()) | |||
saver = tf.train.Saver(var_list=tf.trainable_variables(), max_to_keep=args.max_checkpoints) # TODO hier ansetzen |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
What does this TODO mean?
train.py
Outdated
@@ -44,13 +45,13 @@ def _str_to_bool(s): | |||
|
|||
parser = argparse.ArgumentParser(description='WaveNet example network') | |||
parser.add_argument('--batch_size', type=int, default=BATCH_SIZE, | |||
help='How many wav files to process at once.') | |||
help='How many wav files to process at once. Default: 1') |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This should be taken from BATCH_SIZE variable.
train.py
Outdated
parser.add_argument('--data_dir', type=str, default=DATA_DIRECTORY, | ||
help='The directory containing the VCTK corpus.') | ||
parser.add_argument('--store_metadata', type=bool, default=False, | ||
help='Whether to store advanced debugging information ' | ||
'(execution time, memory consumption) for use with ' | ||
'TensorBoard.') | ||
'TensorBoard. Default: False') |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Let's also use a variable with default parameter value here (and other similar places).
…er.add_argument functions. Also removed two TODOs I forgot to delete. Sorry about that.
What is a use case where having more than few last checkpoints would be useful? |
|
@Fjolnir-Dvorak ok, sounds reasonable. |
added default values to --help and added a new parameter --max_checkpoints to prevent default deleting checkpoints if there are more than 5.