Skip to content

GitLab

Unverified Commit f98ec55e authored by Manoj Plakal's avatar Manoj Plakal Committed by GitHub
Browse files

Merge pull request #3353 from plakal/master 

Explicitly specify checkpoint version.
parents aa2b818e d9f6b6f3
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
research/audioset/vggish_slim.py
View file @ f98ec55e
...@@ -124,5 +124,6 @@ def load_vggish_slim_checkpoint(session, checkpoint_path): ...@@ -124,5 +124,6 @@ def load_vggish_slim_checkpoint(session, checkpoint_path):
vggish_vars = [v for v in tf.global_variables() if v.name in vggish_var_names] vggish_vars = [v for v in tf.global_variables() if v.name in vggish_var_names]
# Use a Saver to restore just the variables selected above. # Use a Saver to restore just the variables selected above.
saver = tf.train.Saver(vggish_vars, name='vggish_load_pretrained') saver = tf.train.Saver(vggish_vars, name='vggish_load_pretrained',
write_version=1)
saver.restore(session, checkpoint_path) saver.restore(session, checkpoint_path)
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