Reload weights after plateau.
This commit is contained in:
parent
eb23728538
commit
09e1422278
1
.gitignore
vendored
1
.gitignore
vendored
@ -37,3 +37,4 @@ Dockerfile.train
|
||||
doc/xml-c
|
||||
doc/xml-java
|
||||
doc/xml-dotnet
|
||||
convert_graphdef_memmapped_format
|
||||
|
@ -645,6 +645,9 @@ def train():
|
||||
log_info('Encountered a plateau, reducing learning rate to {}'.format(
|
||||
current_learning_rate))
|
||||
|
||||
# Reload checkpoint that we use the best_dev weights again
|
||||
load_or_init_graph_for_training(session)
|
||||
|
||||
if FLAGS.metrics_files:
|
||||
# Read only metrics, not affecting best validation loss tracking
|
||||
for source, init_op in zip(metrics_sources, metrics_init_ops):
|
||||
|
Loading…
Reference in New Issue
Block a user